summaryrefslogtreecommitdiff
path: root/training/conf/trainer/default.yaml
blob: c2d0d62989a3310728e1a91aef06bceab81f3d9f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
_target_: pytorch_lightning.Trainer
stochastic_weight_avg: true
auto_scale_batch_size: binsearch
auto_lr_find: false
gradient_clip_val: 0.5
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
resume_from_checkpoint: null
accumulate_grad_batches: 1
overfit_batches: 0