From 109bb74967b3de1a3dadeeafa3485122d6ca7609 Mon Sep 17 00:00:00 2001
From: Mark Holmstrom <mholmstr@ualberta.ca>
Date: Sat, 4 Jan 2025 19:31:18 -0700
Subject: [PATCH] Fix Numpy Random Seed

---
 main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index ad69227..951b6f7 100644
--- a/main.py
+++ b/main.py
@@ -16,6 +16,7 @@ def main():
         torch.manual_seed(args.seed)
         torch.cuda.manual_seed(args.seed)
         torch.cuda.manual_seed_all(args.seed)
+        np.random.seed(args.seed)
         
         
     # Make the a directory corresponding to this run for saving results, checkpoints etc.
@@ -68,4 +69,4 @@ def main():
 
    
 if __name__ == "__main__":
-    main()
\ No newline at end of file
+    main()