blob: ef5b018b4729341559ae680d245ae0873d873570 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
_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: top
limit_train_batches: 1.0
limit_val_batches: 1.0
limit_test_batches: 1.0
resume_from_checkpoint: null
accumulate_grad_batches: 1
|