glenn-jocher commited on
Commit
023e378
1 Parent(s): 64e6d19

hyperparameter evolution update (#566)

Browse files
Files changed (1) hide show
  1. utils/utils.py +1 -1
utils/utils.py CHANGED
@@ -841,7 +841,7 @@ def print_mutation(hyp, results, yaml_file='hyp_evolved.yaml', bucket=''):
841
  for i, k in enumerate(hyp.keys()):
842
  hyp[k] = float(x[0, i + 7])
843
  with open(yaml_file, 'w') as f:
844
- results = x[0, :7]
845
  c = '%10.4g' * len(results) % results # results (P, R, mAP@0.5, mAP@0.5:0.95, val_losses x 3)
846
  f.write('# Hyperparameter Evolution Results\n# Generations: %g\n# Metrics: ' % len(x) + c + '\n\n')
847
  yaml.dump(hyp, f, sort_keys=False)
 
841
  for i, k in enumerate(hyp.keys()):
842
  hyp[k] = float(x[0, i + 7])
843
  with open(yaml_file, 'w') as f:
844
+ results = tuple(x[0, :7])
845
  c = '%10.4g' * len(results) % results # results (P, R, mAP@0.5, mAP@0.5:0.95, val_losses x 3)
846
  f.write('# Hyperparameter Evolution Results\n# Generations: %g\n# Metrics: ' % len(x) + c + '\n\n')
847
  yaml.dump(hyp, f, sort_keys=False)