Commit 56bc7c7c authored by lucas_miranda's avatar lucas_miranda
Browse files

Fixed issue with gmvaep when returning both prior and posterior

parent aafd2ff9
......@@ -328,7 +328,7 @@ def autoencoder_fitting(
return_list = (encoder, decoder, ae)
else:
(encoder, generator, grouper, ae,) = deepof.models.SEQ_2_SEQ_GMVAE(
(encoder, generator, grouper, ae, prior, posterior) = deepof.models.SEQ_2_SEQ_GMVAE(
architecture_hparams=({} if hparams is None else hparams),
batch_size=batch_size,
compile_model=True,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment