steffen/ml_lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Final Train Runs

Browse Source
This commit is contained in:
Steffen Illium 2021-03-18 21:43:26 +01:00
parent fc4617c9d8
commit fff5e6e00a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
utils/tools.py
View File

@ -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):