diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-13 19:09:25 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-13 19:09:25 +0200 |
| commit | 2862fdf77a19c4afa5e00c900af4877df31a3ea6 (patch) | |
| tree | 5a225df3113849b5c129b8ae42497cef383b4e93 /training/conf/trainer | |
| parent | aaa26d7d550bb2d4ced4b87414fd314608148865 (diff) | |
Update configs
Diffstat (limited to 'training/conf/trainer')
| -rw-r--r-- | training/conf/trainer/default.yaml | 3 |
1 files changed, 0 insertions, 3 deletions
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 |