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/