diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-27 16:52:55 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-27 16:52:55 +0200 |
| commit | 0ba945e84d11a07ac95fdf8495f2ff278215adb9 (patch) | |
| tree | b22bcecf37416fbdfbe6a5ef178471e1a5751c3e /training/conf/callbacks/checkpoint.yaml | |
| parent | 22e36513dd43d2e2ca82ca28a1ea757c5663676a (diff) | |
Changed structure of callbacks
Diffstat (limited to 'training/conf/callbacks/checkpoint.yaml')
| -rw-r--r-- | training/conf/callbacks/checkpoint.yaml | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/training/conf/callbacks/checkpoint.yaml b/training/conf/callbacks/checkpoint.yaml new file mode 100644 index 0000000..afc536f --- /dev/null +++ b/training/conf/callbacks/checkpoint.yaml @@ -0,0 +1,5 @@ +type: ModelCheckpoint +args: + monitor: val_loss + mode: min + save_last: true |