diff --git a/bulkmodel.py b/bulkmodel.py index 4e3e2fb..5200ca2 100644 --- a/bulkmodel.py +++ b/bulkmodel.py @@ -34,9 +34,9 @@ def run_main(args): torch.cuda.manual_seed(seed) torch.cuda.manual_seed_all(seed) #from transformers import * - random.seed(42) - np.random.seed(42) - os.environ['PYTHONHASHSEED'] = str(42) + random.seed(seed) + np.random.seed(seed) + os.environ['PYTHONHASHSEED'] = str(seed) #torch.manual_seed(seed) #torch.cuda.manual_seed(seed) torch.backends.cudnn.deterministic = True diff --git a/scmodel.py b/scmodel.py index 07d007e..8218019 100644 --- a/scmodel.py +++ b/scmodel.py @@ -57,9 +57,9 @@ def run_main(args): torch.cuda.manual_seed(seed) torch.cuda.manual_seed_all(seed) #from transformers import * - random.seed(42) - np.random.seed(42) - os.environ['PYTHONHASHSEED'] = str(42) + random.seed(seed) + np.random.seed(seed) + os.environ['PYTHONHASHSEED'] = str(seed) torch.backends.cudnn.deterministic = True torch.backends.cudnn.benchmark=False