From df4f25e610bc31af3ba458dce4e569bb49174745 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 25 Jul 2019 13:51:55 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index a734a6d5af..e25be48316 100644 --- a/train.py +++ b/train.py @@ -340,7 +340,7 @@ def print_mutation(hyp, results): def fitness(x): # returns fitness of hyp evolution vectors - return x[:, 2] * 0.5 + x[:, 3] * 0.5 # fitness = weighted combination of mAP and F1 + return 0.5 * x[:, 2] + 0.5 * x[:, 3] # fitness = 0.5 * mAP + 0.5 * F1 if __name__ == '__main__':