search

bytedance / byteps

A high performance and generic framework for distributed DNN training
Other
3.63k stars 488 forks source link

example: fix import for python3.8 #379

Closed pleasantrabbit closed 3 years ago

pleasantrabbit commented 3 years ago

fix the import statement in pytorch/mnist-distributed.py for python 3.8

fixes https://github.com/bytedance/byteps/issues/376

Signed-off-by: Yulu Jia yulu.jia@bytedance.com