diff --git a/TODO.md b/TODO.md index 4fde24e..c0625c3 100644 --- a/TODO.md +++ b/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 diff --git a/config-new.yaml b/config-new.yaml new file mode 100644 index 0000000..d5550e3 --- /dev/null +++ b/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