data:
  batch_size: 16
  collate: 1
  num_workers: 4
  root_dir: ./data/sample/processed
  test_csv: ./data/csvs/input_scans.csv
  train_csv: ./data/csvs/train_set_100.csv
  val_csv: ./data/csvs/val_set_100.csv
gpu:
  device: cpu
  visible_device: ''
infer:
  checkpoints: ./checkpoints/mci_model.pt
logger:
  project_name: brainage
  run_name: ExperimentName_trainconfigs
  save_dir: ./Checkpoints
  save_name: ExperimentName_trainconfigs_checkpoint-{epoch:02d}-{loss:.2f}-{metric:.2f}
optim:
  clr: 'no'
  lr: 0.0001
  max_epochs: 200
  momentum: 0.9
  weight_decay: 1.0e-05
train:
  finetune: 'yes'
  freeze: 'no'
  weights: path/to/brainiac/weights