Omini3D / Config /config_reg_brain.yaml
maxmo2009's picture
Sync from local: code + epoch-110 checkpoint, clean README
2af0e94 verified
data_name: all
# net_name: recresacnet
# net_name: recmutattnnet
net_name: recmulmodmutattnnet
# net_name: defrecmutattnnet
ndims: 3
img_size: 128
batchsize: 3
ddf_pad_mode: border
device: xpu
img_pad_mode: zeros
num_input_chn: 1
padding_mode: border
resample_mode: bilinear
timesteps: 80
v_scale: 5.0e-05
# =========================
# TRAINING SETTING
epoch: 10000
epoch_per_save: 1
lr: 0.00001
noise_scale: 0.1
# =========================
# AUGMENTATION SETTING
patients_list: []
model_id_str: '000009'
start_noise_step: 75
noise_step: 1
aug_coe: 1
condition_type: 'none'
reg_img_savepath: Data/Reg_data/unpair_brain/img/
reg_msk_savepath: Data/Reg_data/unpair_brain/msk/
reg_ddf_savepath: Data/Reg_data/unpair_brain/ddf/
aug_img_savepath: Data/Aug_data/unpair_brain/img/
aug_msk_savepath: Data/Aug_data/unpair_brain/msk/
aug_ddf_savepath: Data/Aug_data/unpair_brain/ddf/