Skip to content

Commit 215cc26

Browse files
committed
increase n_trials in optimization and lower the batch size to save memory
1 parent 43727d7 commit 215cc26

File tree

1 file changed

+3
-2
lines changed
  • operators/quantem-direct-ptycho

1 file changed

+3
-2
lines changed

operators/quantem-direct-ptycho/run.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -234,15 +234,16 @@ def quantem_direct_ptycho(
234234
aberration_coefs=opt_aberration_coefs,
235235
rotation_angle=opt_rotation_angle,
236236
deconvolution_kernel=deconvolution_kernel,
237-
n_trials=5, # TODO increase this later
237+
n_trials=50,
238+
max_batch_size=10,
238239
)
239240
else:
240241
logger.info(f"Scan {scan_number}: Using manual hyperparameter settings")
241242

242243
initial_parallax = direct_ptycho.reconstruct(
243244
deconvolution_kernel=deconvolution_kernel,
244245
upsampling_factor=upsampling_factor,
245-
max_batch_size=32, # this used to be 10. Lets try 32
246+
max_batch_size=10,
246247
)
247248

248249
# Process and return result

0 commit comments

Comments
 (0)