Register vqvae3
This commit is contained in:
parent
bc20b4739e
commit
29c1c3bede
|
@ -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__':
|
||||||
|
|
Loading…
Reference in New Issue
Block a user