mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Register vqvae3

Browse Source
This commit is contained in:
James Betker 2021-01-29 15:26:28 -07:00
parent bc20b4739e
commit 29c1c3bede
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/vqvae/vqvae_3.py
View File

@ -170,9 +170,9 @@ class VQVAE3(nn.Module):
@register_model @register_model
def register_vqvae_normalized(opt_net, opt): def register_vqvae3(opt_net, opt):
kw = opt_get(opt_net, ['kwargs'], {}) kw = opt_get(opt_net, ['kwargs'], {})
return VQVAE(**kw) return VQVAE3(**kw)
if __name__ == '__main__': if __name__ == '__main__':