blob: d4ffcdcff7ec3591018cfc4c2384c8f997dea3eb (
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: 2
overfit_batches: 0
|
