new yaml schema, new TODO

TODO.md

@@ -1,5 +1,4 @@
 # TODO:
-
 - fold_slice:
     - load diffractions / hdf5 files; change only param per job
     - restructure `FoldSlicePtychoEngine.__init__()` to load data but not params
@@ -17,10 +16,9 @@
 - change job.sub to job.sh
     - dynamic jobname
     - pre-check result directory
-- clean up examples for Si-FIB paper version
 
 # TODAY:
-- rename `BOEngine` to `Sampler`
+- change yaml schema, use stages in config-new.yaml
 - separation of available GPUs and parallel sample batches
 - prepare next batch for efficient GPU use
 

config-new.yaml

@@ -0,0 +1,71 @@
+io:
+  input_data_path: '/home/swim/shared/Si_project/data/Si2V1_2.mat'
+  result_dir: '/home/swim/bo-ptycho/results/test'
+  verbosity: 1
+
+ptycho:
+  engine: 'fake'
+  # path: '/home/swim/fold_slice-park'
+  params:
+    voltage: 200
+    alpha_max: 30
+    defocus: -250
+    rot_ang: 0.3
+    Nlayers: 20
+    thickness: 250
+    rbf: 37
+    tilt_x: 2
+    tilt_y: 0 
+    scan_step_size: 0.36
+
+    Niter: 10
+    Niter_save_results: 10
+
+    CBED_size: 192
+    ADU: 1
+    Nprobe: 1
+    N_scan_x: 64
+    N_scan_y: 64 
+    extra_print_info: 'FIB'
+    scan_number: 1
+    gpu_id: 1
+    roi_label: '0_Ndp64'
+    diff_pattern_blur: 1
+    probe_change_start: 1
+    object_change_start: 1
+    grouping: 64
+    probe_position_search: 1
+    regularize_layers: 0.2
+    variable_probe: false
+
+search:
+  metric: log_fourier
+  
+  params:
+    defocus:
+      radius: 100
+    Nlayers:
+      radius: 5
+      type: int
+    thickness:
+      radius: 150
+  
+  train_x:
+  train_y:
+
+  stages:
+    # - sampler: fixed
+    
+    # - sampler: grid
+    #   defocus: 7
+    #   Nlayers: 11
+    #   thickness: 7
+
+    - sampler: random
+      samples: 16
+
+    - sampler: sobo
+      samples: 256
+      batch: 4
+      acquisition: 'ucb'
+      beta: 0.1