File size: 520 Bytes
97aa5af | 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 | # Default training and data paths for R3PM-Net
exp_name: exp_r3pmnet
eval: false
save_dir: ""
fine_tune_feature_extractor: tune
transfer_weights: "" # Optional: leave empty to skip loading
emb_dims: 1024
symfn: max
seed: 1234
workers: 4
batch_size: 5
epochs: 2
start_epoch: 0
optimizer: Adam
resume: ""
pretrained: ""
device: cuda:0
# Pickled Registration Dataset dicts (keys: template, source, transformation)
train_dict_path: data/simulators/data_dict_train.pkl
test_dict_path: data/simulators/data_dict_test.pkl
|