Update seeds for single-GPU reproducibility (#3789)
For seed=0 on single-GPU.
This commit is contained in:
1 parent
8e7f285051
commit
92d49fde35
1 file changed
+1
-1
@@ -84,7 +84,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
|
|||||||
# Configure
|
# Configure
|
||||||
plots = not evolve # create plots
|
plots = not evolve # create plots
|
||||||
cuda = device.type != 'cpu'
|
cuda = device.type != 'cpu'
|
||||||
init_seeds(2 + RANK)
|
init_seeds(1 + RANK)
|
||||||
with open(data) as f:
|
with open(data) as f:
|
||||||
data_dict = yaml.safe_load(f) # data dict
|
data_dict = yaml.safe_load(f) # data dict
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user