training/conf/trainer/default.yaml


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

_target_: pytorch_lightning.Trainer
stochastic_weight_avg: false
auto_scale_batch_size: binsearch
auto_lr_find: false
gradient_clip_val: 0
fast_dev_run: false
gpus: 1
precision: 16
max_epochs: 512
terminate_on_nan: true
weights_summary: full
limit_train_batches: 1.0 
limit_val_batches: 1.0
limit_test_batches: 1.0
resume_from_checkpoint: null