From 29c1c3bede8609255cb046ac7a0397f208bbb2d9 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Fri, 29 Jan 2021 15:26:28 -0700
Subject: [PATCH] Register vqvae3

---
 codes/models/vqvae/vqvae_3.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/codes/models/vqvae/vqvae_3.py b/codes/models/vqvae/vqvae_3.py
index 3c040775..dc3f92cb 100644
--- a/codes/models/vqvae/vqvae_3.py
+++ b/codes/models/vqvae/vqvae_3.py
@@ -170,9 +170,9 @@ class VQVAE3(nn.Module):
 
 
 @register_model
-def register_vqvae_normalized(opt_net, opt):
+def register_vqvae3(opt_net, opt):
     kw = opt_get(opt_net, ['kwargs'], {})
-    return VQVAE(**kw)
+    return VQVAE3(**kw)
 
 
 if __name__ == '__main__':