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
# 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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论