From 9c9c09ca34a76cde26f5c95ac59ba3cf26f17255 Mon Sep 17 00:00:00 2001 From: Dario Coscia <93731561+dario-coscia@users.noreply.github.com> Date: Tue, 30 Apr 2024 17:19:43 +0200 Subject: [PATCH 1/3] Update label_tensor.py cpu/gpu --- pina/label_tensor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pina/label_tensor.py b/pina/label_tensor.py index fe8e1a850..c8a41f7b4 100644 --- a/pina/label_tensor.py +++ b/pina/label_tensor.py @@ -176,7 +176,7 @@ def cuda(self, *args, **kwargs): tmp = super().cuda(*args, **kwargs) new = self.__class__.clone(self) new.data = tmp.data - return tmp + return new def cpu(self, *args, **kwargs): """ @@ -185,7 +185,7 @@ def cpu(self, *args, **kwargs): tmp = super().cpu(*args, **kwargs) new = self.__class__.clone(self) new.data = tmp.data - return tmp + return new def extract(self, label_to_extract): """ From 347d9011519d5d442b5fb6f67be7155a1deded60 Mon Sep 17 00:00:00 2001 From: Dario Coscia <93731561+dario-coscia@users.noreply.github.com> Date: Tue, 30 Apr 2024 18:17:41 +0200 Subject: [PATCH 2/3] Update test_adaptive_refinment_callbacks.py --- tests/test_callbacks/test_adaptive_refinment_callbacks.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_callbacks/test_adaptive_refinment_callbacks.py b/tests/test_callbacks/test_adaptive_refinment_callbacks.py index fb74367e6..214257d95 100644 --- 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() From 033b6f719e964f8fcba90f334d0c19d6157137f2 Mon Sep 17 00:00:00 2001 From: Dario Coscia <93731561+dario-coscia@users.noreply.github.com> Date: Tue, 30 Apr 2024 18:32:40 +0200 Subject: [PATCH 3/3] Update test_optimizer_callbacks.py --- tests/test_callbacks/test_optimizer_callbacks.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_callbacks/test_optimizer_callbacks.py b/tests/test_callbacks/test_optimizer_callbacks.py index 6c167b600..0b0aabaab 100644 --- 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()