File size: 653 Bytes
5a169ab |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
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
|