From 3de61b1fa597fc01b692316fa6c277b9ea484320 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 22 Jan 2020 11:08:03 -0800
Subject: [PATCH] updates

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 4535c0d5..97bbc773 100644
--- a/train.py
+++ b/train.py
@@ -452,9 +452,9 @@ if __name__ == '__main__':
                 x = x[np.argsort(-fitness(x))][:n]  # top n mutations
                 if parent == 'single' or len(x) == 1:
                     x = x[random.randint(0, n - 1)]  # select one of the top n
-                elif parent == 'weighted':  # weighted combination
+                elif parent == 'weighted':
                     w = fitness(x) - fitness(x).min()  # weights
-                    x = (x * w.reshape(n, 1)).sum(0) / w.sum()  # new parent
+                    x = (x * w.reshape(n, 1)).sum(0) / w.sum()  # select weighted combination
 
                 # Mutate
                 method = 3