add diffusion models

Diffusion/DDPM/config.yaml

@@ -0,0 +1,37 @@
+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

Diffusion/DDPM/ddpm.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4a2aedd3e021b57c7b48e0393ce0209b2fed5d0b3543170f2442a797c8bf9bc3
+size 96274464

Diffusion/DDPM/ddpm.meta

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b33b3f45d2b582b8ddefce008b79da4ba2d5208a002017e62575ba5460a982a9
+size 55628213

Diffusion/SMLD/config.yaml

@@ -0,0 +1,38 @@
+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

Diffusion/SMLD/smld.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6083c82c67f36c94527280787e31831a3a8e8a9e8e5f15113770bf184d1381c2
+size 96274464

Diffusion/SMLD/smld.meta

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dca2f3f7c45e0f0f25fb716a8bb826d1d9c459ac25cc83fded4524b3a81b544f
+size 16669007