Do not save hyp.yaml and opt.yaml on evolve (#5775)

* Do not save hyp.yaml and opt.yaml on evolve

* Update general.py
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-11-24 23:25:51 +01:00
1 parent 7a39803476
commit 4ca4aec46f
2 files changed
+7 -6

No files matched your search

+1 -1
View File
@@ -777,7 +777,7 @@ def print_mutation(results, hyp, save_dir, bucket):
i = np.argmax(fitness(data.values[:, :7])) #
f.write('# YOLOv5 Hyperparameter Evolution Results\n' +
f'# Best generation: {i}\n' +
f'# Last generation: {len(data)}\n' +
f'# Last generation: {len(data) - 1}\n' +
'# ' + ', '.join(f'{x.strip():>20s}' for x in keys[:7]) + '\n' +
'# ' + ', '.join(f'{x:>20.5g}' for x in data.values[i, :7]) + '\n\n')
yaml.safe_dump(hyp, f, sort_keys=False)