Update seeding: add CUDA seeding

zuoxingdong · Sep 9, 2019 · e817053 · e817053
1 parent 6e06813
commit e817053
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lagom/utils/seeding.py b/lagom/utils/seeding.py
@@ -17,7 +17,8 @@ def set_global_seeds(seed):
     """
     random.seed(seed)
     np.random.seed(seed)
-    torch.manual_seed(seed)  # both torch and torch.cuda internally
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed(seed)
 
 
 class Seeder(object):