fix(encoder): fix error in cvae encoder

gnes-ai · Jul 23, 2019 · ab6c88c · ab6c88c
1 parent a4b883a
commit ab6c88c
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 1 deletion.
diff --git a/gnes/encoder/__init__.py b/gnes/encoder/__init__.py
@@ -40,6 +40,7 @@
     'HashEncoder': 'numeric.hash',
     'BasePytorchEncoder': 'image.base',
     'TFInceptionEncoder': 'image.inception',
+    'CVAEEncoder': 'image.cvae'
 }
 
 register_all_class(_cls2file_map, 'encoder')
diff --git a/gnes/encoder/image/cvae.py b/gnes/encoder/image/cvae.py
@@ -40,7 +40,7 @@ def __init__(self, model_dir: str,
 
     def post_init(self):
         import tensorflow as tf
-        from .cave_cores.model import CVAE
+        from .cvae_cores.model import CVAE
 
         self._model = CVAE(self.latent_dim)
         self.inputs = tf.placeholder(tf.float32,

diff --git a/gnes/encoder/image/cvae_cores/__init__.py b/gnes/encoder/image/cvae_cores/__init__.py