Unverified 提交 92d49fde authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update seeds for single-GPU reproducibility (#3789)

For seed=0 on single-GPU.
上级 8e7f2850
...@@ -84,7 +84,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary ...@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论