Skip to content
GitLab
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
86a1c09e
Commit
86a1c09e
authored
Apr 23, 2021
by
lucas_miranda
Browse files
Added warmup parameters to model output naming
parent
38867e7e
Pipeline
#100165
passed with stages
in 23 minutes and 49 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
deepof/train_utils.py
View file @
86a1c09e
...
...
@@ -333,6 +333,8 @@ def autoencoder_fitting(
next_sequence_prediction
=
next_sequence_prediction
,
rule_based_prediction
=
rule_based_prediction
,
loss
=
loss
,
loss_warmup
=
kl_warmup
,
warmup_mode
=
kl_annealing_mode
,
input_type
=
input_type
,
X_val
=
(
X_val
if
X_val
.
shape
!=
(
0
,)
else
None
),
cp
=
save_checkpoints
,
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new 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