From 0d9147392f5defdd4af6041ef5a72f076da055b2 Mon Sep 17 00:00:00 2001
From: Shayan Mousavinia <45814390+ShAmoNiA@users.noreply.github.com>
Date: Wed, 26 Apr 2023 09:18:54 +0330
Subject: [PATCH] Update train.py

Signed-off-by: Shayan Mousavinia <45814390+ShAmoNiA@users.noreply.github.com>
---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 = []