Set manual seed.

yangkky · Oct 18, 2019 · 2446796 · 2446796
1 parent 36a7a09
commit 2446796
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/mnist-mixed.py b/src/mnist-mixed.py
@@ -58,7 +58,7 @@ def train(gpu, args):
         init_method='env://',
         world_size=args.world_size,
         rank=rank)
-
+    torch.manual_seed(0)
     model = ConvNet()
     torch.cuda.set_device(gpu)
     model.cuda(gpu)
@@ -117,4 +117,4 @@ def train(gpu, args):
 
 
 if __name__ == '__main__':
-    main()
+    main()