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