cSAXS/cdtools
0
0
Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-09-09 21:12:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug where fancyptycho test was not being run on GPU

Browse Source
This commit is contained in:
levita_a
2026-04-03 18:59:09 +02:00
parent b6553fd522
commit 8ee8bf9237
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/models/test_fancy_ptycho.py
+2 -2
View File
@@ -97,8 +97,8 @@ def test_lab_ptycho(lab_ptycho_cxi, reconstruction_device, show_plot):
print('Running reconstruction on provided reconstruction_device,',
reconstruction_device)
#model.to(device=reconstruction_device)
#dataset.get_as(device=reconstruction_device)
model.to(device=reconstruction_device)
dataset.get_as(device=reconstruction_device)
for loss in model.Adam_optimize(50, dataset, lr=0.02, batch_size=10):
print(model.report())