Fix accelerator in tests/test_train.py (#595)

ashleve · Aug 24, 2023 · 271972c · 271972c
1 parent 8055898
commit 271972c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_train.py b/tests/test_train.py
@@ -44,7 +44,7 @@ def test_train_epoch_gpu_amp(cfg_train: DictConfig) -> None:
     HydraConfig().set_config(cfg_train)
     with open_dict(cfg_train):
         cfg_train.trainer.max_epochs = 1
-        cfg_train.trainer.accelerator = "cpu"
+        cfg_train.trainer.accelerator = "gpu"
         cfg_train.trainer.precision = 16
     train(cfg_train)