From fff5e6e00a94ada47b58df4210ae004769ea24ea Mon Sep 17 00:00:00 2001 From: Steffen Illium Date: Thu, 18 Mar 2021 21:43:26 +0100 Subject: [PATCH] Final Train Runs --- utils/tools.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/utils/tools.py b/utils/tools.py index 3119f0e..80d0424 100644 --- a/utils/tools.py +++ b/utils/tools.py @@ -25,10 +25,10 @@ def to_one_hot(idx_array, max_classes): return one_hot -def fix_all_random_seeds(config_obj): - np.random.seed(config_obj.main.seed) - torch.manual_seed(config_obj.main.seed) - random.seed(config_obj.main.seed) +def fix_all_random_seeds(seed): + np.random.seed(seed) + torch.manual_seed(seed) + random.seed(seed) def write_to_shelve(file_path, value):