Update train.py

SubmissionsIn · Mar 5, 2023 · df1ad15 · df1ad15
1 parent 361a379
commit df1ad15
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -69,8 +69,8 @@
 def setup_seed(seed):
     torch.manual_seed(seed)
     torch.cuda.manual_seed_all(seed)
-#     np.random.seed(seed)
-#     random.seed(seed)
+    # np.random.seed(seed)
+    # random.seed(seed)
     torch.backends.cudnn.deterministic = True