diff --git a/train.py b/train.py index c3aec49df522..a4d0eeea5708 100644 --- a/train.py +++ b/train.py @@ -678,7 +678,7 @@ def main(opt, callbacks=Callbacks()): keys = ('metrics/precision', 'metrics/recall', 'metrics/mAP_0.5', 'metrics/mAP_0.5:0.95', 'val/box_loss', 'val/obj_loss', 'val/cls_loss') print_mutation(keys, results, hyp.copy(), save_dir, opt.bucket) - fitness_scores.append(1) + fitness_scores.append(results[2]) # Select the fittest individuals for reproduction using adaptive tournament selection selected_indices = []