Sadjad Alikhani commited on
Commit
60ef271
·
verified ·
1 Parent(s): 18582c8

Update inference.py

Browse files
Files changed (1) hide show
  1. 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