From 5f21139623049edeb9c356eddba026464a0bec69 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 17 Apr 2019 19:06:15 +0200 Subject: [PATCH] updates --- train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index 8c386687..d6364fc0 100644 --- a/train.py +++ b/train.py @@ -302,13 +302,13 @@ if __name__ == '__main__': old_hyp = hyp.copy() init_seeds(seed=int(time.time())) for k in hyp.keys(): - x = (np.random.randn(1) * 0.3 + 1) ** 1.1 # plt.hist(x.ravel(), 100) + x = (np.random.randn(1) * 0.2 + 1) ** 1.1 # plt.hist(x.ravel(), 100) hyp[k] = hyp[k] * float(x) # vary by about 30% 1sigma # Apply limits hyp['iou_t'] = np.clip(hyp['iou_t'], 0, 0.90) - hyp['momentum'] = np.clip(hyp['momentum'], 0, 0.98) - hyp['weight_decay'] = np.clip(hyp['weight_decay'], 0, 0.01) + hyp['momentum'] = 0.9 # np.clip(hyp['momentum'], 0, 0.98) + hyp['weight_decay'] = 0.0005 # np.clip(hyp['weight_decay'], 0, 0.01) # Normalize loss components (sum to 1) lcf = ['xy', 'wh', 'cls', 'conf']