speed-reproducibility fix #17

This commit is contained in:
Glenn Jocher committed 2020-06-05 13:07:09 -07:00
1 parent 55ca5c74d2
commit 22d6088205
2 files changed
+6 -3

No files matched your search

+1 -1
View File
@@ -63,7 +63,7 @@ def train(hyp):
weights = opt.weights # initial training weights
# Configure
init_seeds()
init_seeds(1)
with open(opt.data) as f:
data_dict = yaml.load(f, Loader=yaml.FullLoader) # model dict
train_path = data_dict['train']