ml_pipeline/src/config/main.yaml

7 lines
95 B
YAML

app_dir: ${hydra:runtime.cwd}
debug: true
lr: 2e-4
batch_size: 16
num_workers: 0
device: "cpu"