try a test

Lightning-AI · Dec 3, 2020 · 22e3c10 · 22e3c10
1 parent 4d2d8fb
commit 22e3c10
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/trainer/logging/test_logger_connector.py b/tests/trainer/logging/test_logger_connector.py
@@ -82,6 +82,7 @@ def training_step_end(self, *_):
 
     trainer = Trainer(
         accelerator=accelerator,
+        gpus=(2 if accelerator == "dp" else None),
         default_root_dir=tmpdir,
         limit_train_batches=2,
         limit_val_batches=4,