Skip to content
This repository has been archived by the owner. It is now read-only.
Permalink
Browse files
fix(encoder): fix error in cvae encoder
  • Loading branch information
Larryjianfeng committed Jul 23, 2019
1 parent a4b883a commit ab6c88ccfe54ba5f96f09510e97b9658c553c1a9
Showing with 2 additions and 1 deletion.
  1. +1 −0 gnes/encoder/__init__.py
  2. +1 −1 gnes/encoder/image/cvae.py
  3. 0 gnes/encoder/image/cvae_cores/__init__.py
@@ -40,6 +40,7 @@
'HashEncoder': 'numeric.hash',
'BasePytorchEncoder': 'image.base',
'TFInceptionEncoder': 'image.inception',
'CVAEEncoder': 'image.cvae'
}

register_all_class(_cls2file_map, 'encoder')
@@ -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,
Empty file.

0 comments on commit ab6c88c

Please sign in to comment.