diff options
Diffstat (limited to 'training/conf/callbacks')
| -rw-r--r-- | training/conf/callbacks/checkpoint.yaml | 5 | ||||
| -rw-r--r-- | training/conf/callbacks/default.yaml | 14 | ||||
| -rw-r--r-- | training/conf/callbacks/early_stopping.yaml | 5 | ||||
| -rw-r--r-- | training/conf/callbacks/learning_rate_monitor.yaml | 3 | ||||
| -rw-r--r-- | training/conf/callbacks/swa.yaml | 23 |
5 files changed, 20 insertions, 30 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 diff --git a/training/conf/callbacks/default.yaml b/training/conf/callbacks/default.yaml deleted file mode 100644 index 74dc30c..0000000 --- a/training/conf/callbacks/default.yaml +++ /dev/null @@ -1,14 +0,0 @@ -# @package _group_ -- type: ModelCheckpoint - args: - monitor: val_loss - mode: min - save_last: true -- type: LearningRateMonitor - args: - logging_interval: step -# - type: EarlyStopping -# args: -# monitor: val_loss -# mode: min -# patience: 10 diff --git a/training/conf/callbacks/early_stopping.yaml b/training/conf/callbacks/early_stopping.yaml new file mode 100644 index 0000000..caab824 --- /dev/null +++ b/training/conf/callbacks/early_stopping.yaml @@ -0,0 +1,5 @@ +type: EarlyStopping +args: + monitor: val_loss + mode: min + patience: 10 diff --git a/training/conf/callbacks/learning_rate_monitor.yaml b/training/conf/callbacks/learning_rate_monitor.yaml new file mode 100644 index 0000000..003ab7a --- /dev/null +++ b/training/conf/callbacks/learning_rate_monitor.yaml @@ -0,0 +1,3 @@ +type: LearningRateMonitor +args: + logging_interval: step diff --git a/training/conf/callbacks/swa.yaml b/training/conf/callbacks/swa.yaml index 144ad6e..279ca69 100644 --- a/training/conf/callbacks/swa.yaml +++ b/training/conf/callbacks/swa.yaml @@ -1,16 +1,7 @@ -# @package _group_ -- type: ModelCheckpoint - args: - monitor: val_loss - mode: min - save_last: true -- type: StochasticWeightAveraging - args: - swa_epoch_start: 0.8 - swa_lrs: 0.05 - annealing_epochs: 10 - annealing_strategy: cos - device: null -- type: LearningRateMonitor - args: - logging_interval: step +type: StochasticWeightAveraging +args: + swa_epoch_start: 0.8 + swa_lrs: 0.05 + annealing_epochs: 10 + annealing_strategy: cos + device: null |