File size: 809 Bytes
75854b3 | 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 | data_name: lct
net_name: recmutattnnet
# net_name: recresacnet
ndims: 3
img_size: 128 #was 128
batchsize: 2
ddf_pad_mode: border
device: cuda
img_pad_mode: zeros
num_input_chn: 1
padding_mode: border
resample_mode: bilinear
timesteps: 80
v_scale: 4.0e-05
# =========================
# TRAINING SETTING
epoch: 10000
epoch_per_save: 1
lr: 0.00001
noise_scale: 0.1
# =========================
# AUGMENTATION SETTING
patients_list: []
model_id_str: '001157'
start_noise_step: 64
noise_step: 1
aug_coe: 32 # how many times each sample will be augmented
condition_type: 'project' # 'None', 'none', 'adding','independ', 'downsample', 'slice', 'project', 'uncon'
aug_img_savepath: Data/Aug_data/lct/img/
aug_msk_savepath: Data/Aug_data/lct/msk/
aug_ddf_savepath: Data/Aug_data/lct/ddf/
|