From ffa4be4bf4e3758e01d52a9c1f354a05a90b93de Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Thu, 15 Apr 2021 22:05:24 +0200 Subject: Created mappings --- training/configs/image_transformer.yaml | 1 + 1 file changed, 1 insertion(+) (limited to 'training/configs') diff --git a/training/configs/image_transformer.yaml b/training/configs/image_transformer.yaml index 88c05c2..228e53f 100644 --- a/training/configs/image_transformer.yaml +++ b/training/configs/image_transformer.yaml @@ -56,6 +56,7 @@ callbacks: args: monitor: val_loss mode: min + save_last: true - type: StochasticWeightAveraging args: swa_epoch_start: 0.8 -- cgit v1.2.3-70-g09d2