File size: 910 Bytes
2af0e94 | 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 30 31 32 33 34 35 36 37 | 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/
|