From a3af8e86cbb46481c48116389d18389b2ac9ac5f Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Wed, 20 May 2020 18:26:51 -0400 Subject: [PATCH] Update test_trainer_distributed.py --- tests/test_trainer_distributed.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_trainer_distributed.py b/tests/test_trainer_distributed.py index 78bc144312..7111540190 100644 --- a/tests/test_trainer_distributed.py +++ b/tests/test_trainer_distributed.py @@ -9,6 +9,8 @@ # python ./tests/test_trainer_distributed.py # and in single-GPU mode: # CUDA_VISIBLE_DEVICES=0 python ./tests/test_trainer_distributed.py +# and in CPU mode: +# CUDA_VISIBLE_DEVICES=-1 python ./tests/test_trainer_distributed.py #