fix missing import os for ddp

panpan0000 · web-flow · commit 50e5ea375b4d · 2023-10-18T13:24:27.000+08:00
diff --git a/docs/2.1/_sources/notes/ddp.rst.txt b/docs/2.1/_sources/notes/ddp.rst.txt
@@ -28,6 +28,7 @@ updated, and all models on different processes should be exactly the same.
     import torch.multiprocessing as mp
     import torch.nn as nn
     import torch.optim as optim
+    import os
     from torch.nn.parallel import DistributedDataParallel as DDP
 
 
