cSAXS/cdtools
0
0
Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-09-11 22:12:38 +02: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
2021-02-04 18:55:55 -05:00
parent fdca857046
commit d664bf351b
9 changed files with 330 additions and 790 deletions
Download Patch File Download Diff File Expand all files Collapse all files
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')