File size: 1,464 Bytes
75854b3 2af0e94 75854b3 2af0e94 75854b3 2af0e94 75854b3 2af0e94 75854b3 2af0e94 75854b3 2af0e94 75854b3 2af0e94 75854b3 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 38 39 40 41 42 43 44 45 46 47 48 | data_name: all
net_name: om_net
# 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.05
# =========================
# AUGMENTATION SETTING
patients_list: []
# model_id_str: '000000'
# model_id_str: '000180' # before registration training
# model_id_str: '000356'
# model_id_str: '000157'
# model_id_str: '000171'
model_id_str: '000009'
start_noise_step: 75
noise_step: 1
# aug_coe: 32 # how many times each sample will be augmented
aug_coe: 1 # how many times each sample will be augmented
condition_type: 'slice' # 'None', 'none', 'adding','independ', 'downsample', 'slice', 'project', 'uncon'
# aug_img_savepath: Data/Aug_data/totseg/img/
# aug_msk_savepath: Data/Aug_data/totseg/msk/
# aug_ddf_savepath: Data/Aug_data/totseg/ddf/
# aug_img_savepath: Data/Aug_data/om/img/
# aug_msk_savepath: Data/Aug_data/om/msk/
# aug_ddf_savepath: Data/Aug_data/om/ddf/
reg_img_savepath: Data/Reg_data/om/img/
reg_msk_savepath: Data/Reg_data/om/msk/
reg_ddf_savepath: Data/Reg_data/om/ddf/
aug_img_savepath: Data/Aug_data/msd/img/
aug_msk_savepath: Data/Aug_data/msd/msk/
aug_ddf_savepath: Data/Aug_data/msd/ddf/ |