in-context-impersonation
19 строк · 448.0 Байт
1_target_: lightning.pytorch.trainer.Trainer
2
3default_root_dir: ${paths.output_dir}
4
5min_epochs: 1 # prevents early stopping
6max_epochs: 10
7
8accelerator: gpu
9devices: 1
10
11# mixed precision for extra speed-up
12# precision: 16
13
14# perform a validation loop every N training epochs
15check_val_every_n_epoch: 1
16
17# set True to to ensure deterministic results
18# makes training slower but gives more reproducibility than just setting seeds
19deterministic: True
20