File size: 739 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 39 |
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
|