Update test_pinn.py

mathLab · Nov 6, 2023 · 4cd4d52 · 4cd4d52
1 parent 32db54d
commit 4cd4d52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_solvers/test_pinn.py b/tests/test_solvers/test_pinn.py
@@ -97,7 +97,7 @@ def test_train_cpu():
     n = 10
     poisson_problem.discretise_domain(n, 'grid', locations=boundaries)
     pinn = PINN(problem = poisson_problem, model=model, extra_features=None, loss=LpLoss())
-    trainer = Trainer(solver=pinn, max_epochs=1, accelerator='cuda', batch_size=20)
+    trainer = Trainer(solver=pinn, max_epochs=1, accelerator='cpu', batch_size=20)
     trainer.train()
 
 def test_train_restore():