Update test_trainer_distributed.py

2020-05-20 18:26:51 -04:00
parent eacea530c1
commit a3af8e86cb
1 changed files with 2 additions and 0 deletions
--- 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
 #