torch.manual_seed also seeds all CUDA devices

Cysu · Aug 28, 2017 · a1df21b · a1df21b
1 parent 0c89e7d
commit a1df21b
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 3 deletions.
diff --git a/examples/oim_loss.py b/examples/oim_loss.py
@@ -71,7 +71,6 @@ def get_data(name, split_id, data_dir, height, width, batch_size, workers,
 def main(args):
     np.random.seed(args.seed)
     torch.manual_seed(args.seed)
-    torch.cuda.manual_seed_all(args.seed)
     cudnn.benchmark = True
 
     # Redirect print to both console and log file

diff --git a/examples/softmax_loss.py b/examples/softmax_loss.py
@@ -70,7 +70,6 @@ def get_data(name, split_id, data_dir, height, width, batch_size, workers,
 def main(args):
     np.random.seed(args.seed)
     torch.manual_seed(args.seed)
-    torch.cuda.manual_seed_all(args.seed)
     cudnn.benchmark = True
 
     # Redirect print to both console and log file

diff --git a/examples/triplet_loss.py b/examples/triplet_loss.py
@@ -73,7 +73,6 @@ def get_data(name, split_id, data_dir, height, width, batch_size, num_instances,
 def main(args):
     np.random.seed(args.seed)
     torch.manual_seed(args.seed)
-    torch.cuda.manual_seed_all(args.seed)
     cudnn.benchmark = True
 
     # Redirect print to both console and log file