Sadjad Alikhani
commited on
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -27,7 +27,7 @@ warnings.filterwarnings('ignore')
|
|
27 |
def set_random_seed(seed=42):
|
28 |
torch.manual_seed(seed)
|
29 |
np.random.seed(seed)
|
30 |
-
random.seed(seed)
|
31 |
if torch.cuda.is_available():
|
32 |
torch.cuda.manual_seed_all(seed)
|
33 |
# Ensures deterministic behavior
|
|
|
27 |
def set_random_seed(seed=42):
|
28 |
torch.manual_seed(seed)
|
29 |
np.random.seed(seed)
|
30 |
+
#random.seed(seed)
|
31 |
if torch.cuda.is_available():
|
32 |
torch.cuda.manual_seed_all(seed)
|
33 |
# Ensures deterministic behavior
|