multi-gpu batched bo

config.yaml

@@ -1,11 +1,11 @@
 job:
-  type: "test"
-  random_iters: 5
-  sobo_iters: 10
+  type: "random+sobo"
+  random_iters: 8
+  sobo_iters: 128
 
 io:
-  input_data_path: '/home/swim/shared/Si_project/data/Si2V1_1.mat'
-  result_dir: '/home/swim/bo-ptycho/results/test'
+  input_data_path: '/home/swim/shared/Si_project/data/Si2V1_2.mat'
+  result_dir: '/home/swim/bo-ptycho/results/260812-Si/Si2V1_2'
   verbosity: 1
 
 ptycho:
@@ -16,7 +16,7 @@ ptycho:
     alpha_max: 30
     defocus: -200
     rot_ang: 0.3
-    Nlayers: 5
+    Nlayers: 20
     thickness: 250
     rbf: 37
     Nprobe: 1
@@ -25,8 +25,10 @@ ptycho:
     tilt_x: 2
     tilt_y: 0 
     scan_step_size: 0.36
-    Niter: 5
-    Niter_save_results: 5
+
+    Niter: 100
+    Niter_save_results: 100
+
     CBED_size: 192
     ADU: 1
     extra_print_info: ''
@@ -36,11 +38,10 @@ ptycho:
     diff_pattern_blur: 1
     probe_change_start: 1
     object_change_start: 1
-    grouping: inf
+    grouping: 512
     probe_position_search: 1
     regularize_layers: 0.2
     variable_probe: 'false'
-    verbosity
 
 bo:
   batch: 4
@@ -50,12 +51,12 @@ bo:
   params:
     alpha_max:
     defocus:
-      # radius: 100
+      radius: 100
     rot_ang:
     Nlayers:
-      radius: 2
+      radius: 5
       type: int
     thickness:
-      # radius: 100
+      radius: 100
   train_x:
   train_y: