Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Lucas Miranda
deepOF
Commits
56bc7c7c
Commit
56bc7c7c
authored
Apr 12, 2021
by
lucas_miranda
Browse files
Fixed issue with gmvaep when returning both prior and posterior
parent
aafd2ff9
Changes
1
Hide whitespace changes
Inline
Side-by-side
deepof/train_utils.py
View file @
56bc7c7c
...
...
@@ -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
,
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment