summaryrefslogtreecommitdiff
path: root/training/conf/trainer/default.yaml
blob: 5ed65523f3b13b935ec919cd89a66024874e8e8d (plain)
1
2
3
4
5
6
7
8
9
10
11
_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: 64
terminate_on_nan: true
weights_summary: top