digital-forensics-labs/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-04-10 12:33:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

updated basic tutorials, better comments, code revision, checked it works with latest pytorch version

Browse Source
This commit is contained in:
Aladdin Persson
2022-12-19 23:39:48 +01:00
parent 3f53d68c4f
commit cd607c395c
14 changed files with 162 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
ML/Pytorch/Basics/set_deterministic_behavior/pytorch_set_seeds.py
View File

@@ -1,7 +1,11 @@
import random, torch, os, numpy as np
import random
import torch
import os
import numpy as np
def seed_everything(seed=42):
os.environ['PYTHONHASHSEED'] = str(seed)
os.environ["PYTHONHASHSEED"] = str(seed)
random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed)
@@ -10,6 +14,7 @@ def seed_everything(seed=42):
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
seed_everything()
# Do training etc after running seed_everything
# Do training etc after running seed_everything