Add title option to plot_image and wrappers; misc fixes

- Add title parameter to plot_image, plot_real, plot_imag, plot_amplitude,
  plot_phase, and plot_colorized
- Various fixes to base.py and fancy_ptycho.py

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
allevitanandClaude Sonnet 4.6 committed 2026-03-20 17:20:52 +01:00
1 parent dd744a00a4
commit f4260837bc
4 files changed
+145 -76

No files matched your search

+1 -4
View File
@@ -14,9 +14,7 @@ model = cdtools.models.FancyPtycho.from_dataset(
propagation_distance=5e-3, # Propagate the initial probe guess by 5 mm
units='mm', # Set the units for the live plots
obj_view_crop=-50, # Expands the field of view in the object plot by 50 pix,
exponentiate_obj=False,
panel_plot_mode=True,
plot_level=2,
panel_plot_mode=True, # Organizes the live plots into panels
)
if t.cuda.is_available():
@@ -30,7 +28,6 @@ if t.cuda.is_available():
# e.g. estimates of the moments of individual parameters
recon = cdtools.reconstructors.AdamReconstructor(model, dataset)
# The learning rate parameter sets the alpha for Adam.
# The beta parameters are (0.9, 0.999) by default
# The batch size sets the minibatch size