gnes.encoder.image.cvae module¶
-
class
gnes.encoder.image.cvae.
CVAEEncoder
(model_dir, latent_dim=300, select_method='MEAN', l2_normalize=False, use_gpu=True, *args, **kwargs)[source]¶ Bases:
gnes.encoder.base.BaseImageEncoder
-
batch_size
= 64¶
-
train
(*args, **kwargs)¶ Train the model, need to be overrided
-