Improve pytorch lightning examples

This commit is contained in:
Alex Burlacu
2023-06-12 15:14:13 +03:00
parent 4ab46dd92c
commit e6f9ef5806
3 changed files with 5 additions and 3 deletions

View File

@@ -8,6 +8,6 @@ trainer:
filename: best
save_last: False
save_top_k: 1
monitor: loss
monitor: train_loss
mode: min
max_epochs: 10