Upgrade PyTorch Lightning to 1.0.2 (#7852)

Co-authored-by: Sam Shleifer <sshleifer@gmail.com>
This commit is contained in:
Sean Naren
2020-10-28 18:59:14 +00:00
committed by GitHub
parent 1b6c8d4811
commit 5e24982e58
8 changed files with 11 additions and 13 deletions

View File

@@ -102,7 +102,6 @@ def get_checkpoint_callback(output_dir, metric, save_top_k=1, lower_is_better=Fa
monitor=f"val_{metric}",
mode="min" if "loss" in metric else "max",
save_top_k=save_top_k,
period=0, # maybe save a checkpoint every time val is run, not just end of epoch.
)
return checkpoint_callback