mirror of
https://github.com/cdtools-developers/cdtools.git
synced 2026-10-29 03:11:31 +01:00
Fix bug where fancyptycho test was not being run on GPU
This commit is contained in:
1 parent
b6553fd522
commit
8ee8bf9237
1 file changed
+2
-2
@@ -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())
|
||||
|
||||
Reference in new issue
Block a user