File size: 377 Bytes
f15d29e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | hydra:
run:
dir: ${oc.env:OUTPUT_DIR,outputs/singlerun/${now:%Y-%m-%d}/${now:%H-%M-%S}}
defaults:
- data_module: mp_20
- trainer: default
- lightning_module: default
- adapter: default
trainer:
max_epochs: 200
logger:
job_type: train_finetune # override default defined in defaults.trainer yaml file
lightning_module:
optimizer_partial:
lr: 5e-6
|