default.yaml 142 Bytes
Newer Older
Tri Dao's avatar
Tri Dao committed
1
2
3
4
5
6
7
_target_: pytorch_lightning.Trainer

# set `gpu` to train on GPU, null to train on CPU only
accelerator: null

min_epochs: 1
max_epochs: 1000