From 02d6b2f9c5a5fcdb62619364b53b11494372e42d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 18 Apr 2019 19:25:41 +0200 Subject: [PATCH] updates --- train.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/train.py b/train.py index 5ab7aecc..a3899767 100644 --- a/train.py +++ b/train.py @@ -11,16 +11,16 @@ from utils.datasets import * from utils.utils import * # Hyperparameters -hyp = {'k': 10.11, # loss multiple - 'xy': 0.1509, # xy loss fraction - 'wh': 0.008077, # wh loss fraction - 'cls': 0.01082, # cls loss fraction - 'conf': 0.8302, # conf loss fraction - 'iou_t': 0.05892, # iou target-anchor training threshold - 'lr0': 0.001475, # initial learning rate - 'lrf': -3.371, # final learning rate = lr0 * (10 ** lrf) - 'momentum': 0.8733, # SGD momentum - 'weight_decay': 0.0006636, # optimizer weight decay +hyp = {'k': 7.789, # loss multiple + 'xy': 0.1966, # xy loss fraction + 'wh': 0.01144, # wh loss fraction + 'cls': 0.01746, # cls loss fraction + 'conf': 0.7745, # conf loss fraction + 'iou_t': 0.05732, # iou target-anchor training threshold + 'lr0': 0.001467, # initial learning rate + 'lrf': -3.904, # final learning rate = lr0 * (10 ** lrf) + 'momentum': 0.9008, # SGD momentum + 'weight_decay': 0.0007289, # optimizer weight decay }