Skip to content

Commit

Permalink
updates
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Jul 25, 2019
1 parent 666d772 commit df4f25e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__':
Expand Down

0 comments on commit df4f25e

Please sign in to comment.