zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

hyperparameter evolution update (#566)

This commit is contained in:
Glenn Jocher 2020-08-02 13:18:03 -07:00
parent 64e6d199ba
commit 023e37807c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/utils.py
View File

@ -841,7 +841,7 @@ def print_mutation(hyp, results, yaml_file='hyp_evolved.yaml', bucket=''):
for i, k in enumerate(hyp.keys()):
hyp[k] = float(x[0, i + 7])
with open(yaml_file, 'w') as f:
results = x[0, :7]
results = tuple(x[0, :7])
c = '%10.4g' * len(results) % results # results (P, R, mAP@0.5, mAP@0.5:0.95, val_losses x 3)
f.write('# Hyperparameter Evolution Results\n# Generations: %g\n# Metrics: ' % len(x) + c + '\n\n')
yaml.dump(hyp, f, sort_keys=False)