From 4ca4aec46fa3ed89e5a16f09f6c85d40380ebb0f Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 24 Nov 2021 23:25:51 +0100 Subject: [PATCH] Do not save hyp.yaml and opt.yaml on evolve (#5775) * Do not save hyp.yaml and opt.yaml on evolve * Update general.py --- train.py | 11 ++++++----- utils/general.py | 2 +- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/train.py b/train.py index 2838936d2d78..8d35f50afb11 100644 --- a/train.py +++ b/train.py @@ -76,13 +76,14 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary LOGGER.info(colorstr('hyperparameters: ') + ', '.join(f'{k}={v}' for k, v in hyp.items())) # Save run settings - with open(save_dir / 'hyp.yaml', 'w') as f: - yaml.safe_dump(hyp, f, sort_keys=False) - with open(save_dir / 'opt.yaml', 'w') as f: - yaml.safe_dump(vars(opt), f, sort_keys=False) - data_dict = None + if not evolve: + with open(save_dir / 'hyp.yaml', 'w') as f: + yaml.safe_dump(hyp, f, sort_keys=False) + with open(save_dir / 'opt.yaml', 'w') as f: + yaml.safe_dump(vars(opt), f, sort_keys=False) # Loggers + data_dict = None if RANK in [-1, 0]: loggers = Loggers(save_dir, weights, opt, hyp, LOGGER) # loggers instance if loggers.wandb: diff --git a/utils/general.py b/utils/general.py index fa56ed49aba8..5a2bcc3660f6 100755 --- a/utils/general.py +++ b/utils/general.py @@ -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)