From 0ba945e84d11a07ac95fdf8495f2ff278215adb9 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 27 Jun 2021 16:52:55 +0200 Subject: Changed structure of callbacks --- training/conf/callbacks/swa.yaml | 23 +++++++---------------- 1 file changed, 7 insertions(+), 16 deletions(-) (limited to 'training/conf/callbacks/swa.yaml') 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 -- cgit v1.2.3-70-g09d2