Skip to content
Permalink
Browse files

fix(encoder): fix error in cvae encoder

  • Loading branch information...
Larryjianfeng committed Jul 23, 2019
1 parent a4b883a commit ab6c88ccfe54ba5f96f09510e97b9658c553c1a9
@@ -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,
No changes.

0 comments on commit ab6c88c

Please sign in to comment.
You can’t perform that action at this time.