Update train.py

Signed-off-by: Shayan Mousavinia <45814390+ShAmoNiA@users.noreply.github.com>
ultralytics · Apr 26, 2023 · 0d91473 · 0d91473
1 parent 91218fc
commit 0d91473
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 = []