examples/seq2seq/test_seq2seq_examples_multi_gpu.py · 5423f2a9d47e7465a2b5cd916e4ad234a4668206 · chenpangpang / transformers

[testing] port test_trainer_distributed to distributed pytest + TestCasePlus enhancements (#8107) · 5423f2a9

Stas Bekman authored Oct 28, 2020



* move the helper code into testing_utils

* port test_trainer_distributed to work with pytest

* improve docs

* simplify notes

* doc

* doc

* style

* doc

* further improvements

* torch might not be available

* real fix

* Apply suggestions from code review
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

5423f2a9

test_seq2seq_examples_multi_gpu.py 6.13 KB

Replace test_seq2seq_examples_multi_gpu.py