Sadjad Alikhani commited on
Commit
18582c8
1 Parent(s): d8bd707

Update input_preprocess.py

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