Commit 206d10d4 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent 454cbfa9
......@@ -368,7 +368,9 @@ else:
tensorboard_callback,
kl_warmup_callback,
mmd_warmup_callback,
tf.keras.callbacks.EarlyStopping("val_mae", patience=5),
tf.keras.callbacks.EarlyStopping(
"val_mae", patience=5, restore_best_weights=True
),
cp_callback,
],
)
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