glenn-jocher commited on
Commit
92d49fd
1 Parent(s): 8e7f285

Update seeds for single-GPU reproducibility (#3789)

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -84,7 +84,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
84
  # Configure
85
  plots = not evolve # create plots
86
  cuda = device.type != 'cpu'
87
- init_seeds(2 + RANK)
88
  with open(data) as f:
89
  data_dict = yaml.safe_load(f) # data dict
90
 
 
84
  # Configure
85
  plots = not evolve # create plots
86
  cuda = device.type != 'cpu'
87
+ init_seeds(1 + RANK)
88
  with open(data) as f:
89
  data_dict = yaml.safe_load(f) # data dict
90