libokj's picture
Update configs/trainer/default.yaml
96a31ab
raw
history blame
547 Bytes
_target_: lightning.Trainer
default_root_dir: ${paths.output_dir}
min_epochs: 1
max_epochs: 50
accelerator: auto
precision: 32
gradient_clip_val: 0.5
gradient_clip_algorithm: norm
# deterministic algorithms might make training slower but offers more reproducibility than only setting seeds
# True: use deterministic always, throwing an error on an operation that doesn't support deterministic
# warn: use deterministic when possible, throwing warnings on operations that don’t support deterministic
deterministic: warn
inference_mode: True