DPACMAN / configs /trainer /default.yaml
svincoff's picture
updates
4c4b1fc
raw
history blame contribute delete
507 Bytes
_target_: lightning.pytorch.trainer.Trainer
default_root_dir: ${paths.output_dir}
min_epochs: 1 # prevents early stopping
max_epochs: 10
accelerator: cpu
devices: 1
#gradient_clip_val: 1.0
#gradient_clip_algorithm: "norm"
# mixed precision for extra speed-up
# precision: 16
# perform a validation loop every N training epochs
check_val_every_n_epoch: 1
# set True to to ensure deterministic results
# makes training slower but gives more reproducibility than just setting seeds
deterministic: False