requirements updated and params adjusted

2020-05-21 12:15:10 +02:00
parent 7dd10d9a14
commit 4a975e4f4d
3 changed files with 43 additions and 30 deletions
--- a/main.py
+++ b/main.py
@ -23,9 +23,9 @@ def fix_all_random_seeds(config_obj):
    import numpy as np
    import torch
    import random
-    np.random.seed(config.main.seed)
-    torch.manual_seed(config.main.seed)
-    random.seed(config.main.seed)
+    np.random.seed(config_obj.main.seed)
+    torch.manual_seed(config_obj.main.seed)
+    random.seed(config_obj.main.seed)


 def run_lightning_loop(config_obj):