| data_name: all | |
| # net_name: recresacnet | |
| # net_name: recmutattnnet | |
| net_name: recmutattnnet_contrastive | |
| # net_name: recmutattnnet1 | |
| # net_name: defrecmutattnnet | |
| ndims: 3 | |
| img_size: 128 | |
| batchsize: 1 #1 for testing | |
| ddf_pad_mode: border | |
| device: cuda | |
| img_pad_mode: zeros | |
| num_input_chn: 1 | |
| padding_mode: border | |
| resample_mode: bilinear | |
| timesteps: 80 | |
| v_scale: 5.0e-05 | |
| # ========================= | |
| # TRAINING SETTING | |
| epoch: 100 | |
| epoch_per_save: 1 | |
| lr: 0.00001 | |
| noise_scale: 0.1 | |
| # ========================= | |
| # AUGMENTATION SETTING | |
| patients_list: [] | |
| # model_id_str: '000000' | |
| # model_id_str: '000180' # before registration training | |
| # model_id_str: '000353' # good augmentation results on msd | |
| model_id_str: '000354' # | |
| # model_id_str: '000157' | |
| # model_id_str: '000171' | |
| start_noise_step: 48 # starting from which noise step to add noise | |
| noise_step: 1 | |
| aug_coe: 64 # how many times each sample will be augmented | |
| # start_noise_step: 56 # starting from which noise step to add noise | |
| # noise_step: 4 | |
| # aug_coe: 4 # how many times each sample will be augmented | |
| condition_type: 'uncon' # '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/ |