example: fix import for python3.8 (#379)

fix the import statement in pytorch/mnist-distributed.py for python 3.8 Signed-off-by: Yulu Jia <yulu.jia@bytedance.com>
bytedance · Mar 30, 2021 · 47ff669 · 47ff669
1 parent 85798db
commit 47ff669
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/example/pytorch/mnist-distributed.py b/example/pytorch/mnist-distributed.py
@@ -7,7 +7,7 @@
 import torch
 import torch.nn as nn
 import torch.distributed as dist
-from nn.parallel import DistributedDataParallel as DDP
+from torch.nn.parallel import DistributedDataParallel as DDP
 
 
 def main():