| 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/ |