cSAXS/cdtools
0
0 Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-11-10 00:04:01 +01:00
Code Issues Packages Projects Releases Wiki Activity

Remove the original, worse unified mode model, and update the plotting functions to handle stacks of images, very helpful for multislice and also multi-mode models

Browse Source
This commit is contained in:
Abe Levitan committed 2021-02-04 18:55:55 -05:00
1 parent fdca857046
commit d664bf351b
9 files changed
+330 -790
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

examples/gold_ball_ptycho.py
+2 -2
View File
@@ -17,8 +17,8 @@ model = CDTools.models.FancyPtycho.from_dataset(dataset, n_modes=2)
model.to(device='cuda')
dataset.get_as(device='cuda')
#for i, loss in enumerate(model.Adam_optimize(20, dataset, batch_size=50)):
for i, loss in enumerate(model.LBFGS_optimize(20, dataset, lr=1, history_size=5)):
for i, loss in enumerate(model.Adam_optimize(20, dataset, batch_size=50)):
#for i, loss in enumerate(model.LBFGS_optimize(20, dataset, lr=1, history_size=5)):
# And we liveplot the updates to the model as they happen
print(i,loss)
model.inspect(dataset)
examples/unified_modes.py
+1 -2
View File
@@ -10,8 +10,7 @@ dataset = CDTools.datasets.Ptycho2DDataset.from_cxi(filename)
# dataset.inspect()
# plt.show()
#model = CDTools.models.UnifiedModePtycho.from_dataset(dataset, oversampling=2,n_modes=3)#, probe_support_radius=90)
model = CDTools.models.UnifiedModePtycho2.from_dataset(dataset, oversampling=1,n_modes=3)#, probe_support_radius=90)
model = CDTools.models.UnifiedModePtycho.from_dataset(dataset, oversampling=1,n_modes=3, dm_rank=-1)#, probe_support_radius=90)
#model = CDTools.models.FancyPtycho.from_dataset(dataset, oversampling=1)
model.to(device='cuda')