From 64e6d199ba05cd1746879188eaf519aa4057cb7d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 2 Aug 2020 13:17:26 -0700 Subject: [PATCH] hyperparameter evolution update (#566) --- utils/utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/utils/utils.py b/utils/utils.py index a8be00d..2e6dad7 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -841,6 +841,8 @@ 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] + 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)