SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update test_trainer_distributed.py

Browse Source
This commit is contained in:
Julien Chaumond
2020-05-20 18:26:51 -04:00
parent eacea530c1
commit a3af8e86cb
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_trainer_distributed.py
View File

@@ -9,6 +9,8 @@
# python ./tests/test_trainer_distributed.py # python ./tests/test_trainer_distributed.py
# and in single-GPU mode: # and in single-GPU mode:
# CUDA_VISIBLE_DEVICES=0 python ./tests/test_trainer_distributed.py # CUDA_VISIBLE_DEVICES=0 python ./tests/test_trainer_distributed.py
# and in CPU mode:
# CUDA_VISIBLE_DEVICES=-1 python ./tests/test_trainer_distributed.py
# #