Force non-zero hyp evolution weights w (#3748)
Fix for https://github.com/ultralytics/yolov5/issues/3741
This commit is contained in:
1 parent
fdc22398fa
commit
ae4261c774
1 file changed
+1
-1
@@ -608,7 +608,7 @@ def main(opt):
|
||||
x = np.loadtxt('evolve.txt', ndmin=2)
|
||||
n = min(5, len(x)) # number of previous results to consider
|
||||
x = x[np.argsort(-fitness(x))][:n] # top n mutations
|
||||
w = fitness(x) - fitness(x).min() # weights
|
||||
w = fitness(x) - fitness(x).min() + 1E-6 # weights (sum > 0)
|
||||
if parent == 'single' or len(x) == 1:
|
||||
# x = x[random.randint(0, n - 1)] # random selection
|
||||
x = x[random.choices(range(n), weights=w)[0]] # weighted selection
|
||||
|
||||
Reference in new issue
Block a user