Update seeds for single-GPU reproducibility (#3789)

For seed=0 on single-GPU.
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-06-26 15:42:40 +02:00
1 parent 8e7f285051
commit 92d49fde35
1 file changed
+1 -1
+1 -1
View File
@@ -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