Update label_tensor.py cpu/gpu (#292)

* Update label_tensor.py cpu/gpu * Update test_adaptive_refinment_callbacks.py * Update test_optimizer_callbacks.py
2024-04-30 18:52:09 +02:00
parent 7e38fb3a82
commit 0e98abf204
3 changed files with 4 additions and 2 deletions
--- a/tests/test_callbacks/test_adaptive_refinment_callbacks.py
+++ b/tests/test_callbacks/test_adaptive_refinment_callbacks.py
@@ -71,6 +71,7 @@ def test_r3refinment_routine():
    # make the trainer
    trainer = Trainer(solver=solver,
                      callbacks=[R3Refinement(sample_every=1)],
+                      accelerator='cpu',
                      max_epochs=5)
    trainer.train()

--- a/tests/test_callbacks/test_optimizer_callbacks.py
+++ b/tests/test_callbacks/test_optimizer_callbacks.py
@@ -84,5 +84,6 @@ def test_switch_optimizer_routine():
                                          new_optimizers_kwargs={'lr': 0.01},
                                          epoch_switch=3)
                      ],
+                      accelerator='cpu',
                      max_epochs=5)
    trainer.train()