Fix DDP bug in single process multiple device use cases

This commit is contained in:
Lornatang committed 2020-06-20 13:00:03 +08:00
1 parent 0a09882e4f
commit 899f1d4bde
2 files changed
+2 -1

No files matched your search

+1 -1
View File
@@ -2,7 +2,7 @@
Cython
numpy==1.17
opencv-python
torch>=1.5
torch==1.4
matplotlib
pillow
tensorboard