Skip to content

Commit

Permalink
Update seeds for single-GPU reproducibility (#3789)
Browse files Browse the repository at this point in the history
For seed=0 on single-GPU.
  • Loading branch information
glenn-jocher authored Jun 26, 2021
1 parent 8e7f285 commit 92d49fd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
# Configure
plots = not evolve # create plots
cuda = device.type != 'cpu'
init_seeds(2 + RANK)
init_seeds(1 + RANK)
with open(data) as f:
data_dict = yaml.safe_load(f) # data dict

Expand Down

0 comments on commit 92d49fd

Please sign in to comment.