Sadjad Alikhani
commited on
Commit
•
18582c8
1
Parent(s):
d8bd707
Update input_preprocess.py
Browse files- input_preprocess.py +1 -1
input_preprocess.py
CHANGED
@@ -19,7 +19,7 @@ import torch
|
|
19 |
def set_random_seed(seed=42):
|
20 |
torch.manual_seed(seed)
|
21 |
np.random.seed(seed)
|
22 |
-
random.seed(seed)
|
23 |
if torch.cuda.is_available():
|
24 |
torch.cuda.manual_seed_all(seed)
|
25 |
# Ensures deterministic behavior
|
|
|
19 |
def set_random_seed(seed=42):
|
20 |
torch.manual_seed(seed)
|
21 |
np.random.seed(seed)
|
22 |
+
#random.seed(seed)
|
23 |
if torch.cuda.is_available():
|
24 |
torch.cuda.manual_seed_all(seed)
|
25 |
# Ensures deterministic behavior
|