diff --git a/src/mnist-mixed.py b/src/mnist-mixed.py index 124cc71..1425082 100644 --- 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() \ No newline at end of file + main()