Commit 86a1c09e authored by lucas_miranda's avatar lucas_miranda
Browse files

Added warmup parameters to model output naming

parent 38867e7e
Pipeline #100165 passed with stages
in 23 minutes and 49 seconds
......@@ -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,
......
Markdown is supported
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