From 2862fdf77a19c4afa5e00c900af4877df31a3ea6 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 13 Sep 2022 19:09:25 +0200 Subject: Update configs --- training/conf/trainer/default.yaml | 3 --- 1 file changed, 3 deletions(-) (limited to 'training/conf/trainer/default.yaml') diff --git a/training/conf/trainer/default.yaml b/training/conf/trainer/default.yaml index c2d0d62..6112cd8 100644 --- a/training/conf/trainer/default.yaml +++ b/training/conf/trainer/default.yaml @@ -1,5 +1,4 @@ _target_: pytorch_lightning.Trainer -stochastic_weight_avg: true auto_scale_batch_size: binsearch auto_lr_find: false gradient_clip_val: 0.5 @@ -7,8 +6,6 @@ fast_dev_run: false gpus: 1 precision: 16 max_epochs: 256 -terminate_on_nan: true -weights_summary: null limit_train_batches: 1.0 limit_val_batches: 1.0 limit_test_batches: 1.0 -- cgit v1.2.3-70-g09d2