Update seeds for single-GPU reproducibility (#3789)
Browse files
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(
|
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 |
|