| model: SDE | |
| batch_size: 10 | |
| input_shape: | |
| - 256 | |
| - 256 | |
| - 2 | |
| data_chns: 2 | |
| s_type: quad | |
| sigma_max: 5.0 | |
| sigma_min: 0.001 | |
| N: 100 | |
| eps: 1.0e-05 | |
| reduce_mean: true | |
| lr_warm_up_steps: 1000 | |
| lr_start: 0.0001 | |
| lr_min: 0.0001 | |
| lr_max: 0.0001 | |
| lr_max_decay_steps: 2000 | |
| seed: 1234 | |
| net: refine | |
| body: small | |
| nr_filters: 64 | |
| nonlinearity: elu | |
| fourier_scale: 16 | |
| affine_x: false | |
| attention: true | |
| max_keep: 100 | |
| max_epochs: 50 | |
| save_interval: 5 | |
| saved_name: sde_abide | |
| log_folder: /home/gluo/workspace/nlinv_prior/logs | |
| restore_path: /home/gluo/workspace/nlinv_prior/logs/20230330-214422/sde_abide_50 | |
| num_thread: 30 | |
| print_loss: true | |
| train_list: /home/gluo/workspace/nlinv_prior/data/abide/train | |
| test_list: /home/gluo/workspace/nlinv_prior/data/abide/test | |
| nr_gpu: 2 | |
| gpu_id: 0,1 | |