File size: 633 Bytes
2fe61a5 |
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 |
model: SDE2
batch_size: 10
input_shape:
- 256
- 256
- 2
data_chns: 2
sigma_max: 20.0
sigma_min: 0.1
eps: 1.0e-05
reduce_mean: true
N: 1000
continuous: false
lr_warm_up_steps: 1000
lr_start: 0.0003
lr_min: 0.0003
lr_max: 0.0003
lr_max_decay_steps: 2000
seed: 1234
net: refine
body: small
nr_filters: 64
nonlinearity: elu
fourier_scale: 16
affine_x: false
attention: false
max_keep: 100
max_epochs: 3000
save_interval: 50
saved_name: sde2_abide
log_folder: /scratch/users/luo9/abide
num_thread: 30
print_loss: true
train_list: /scratch/users/luo9/abide/train
test_list: /scratch/users/luo9/abide/test
nr_gpu: 8
gpu_id: 0,1,2,3,4,5,6,7
|