From 8ae16adde4ed57dc8746a3fe381a27a815167039 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 22 Jul 2019 16:14:41 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index c455d868..14ae5883 100644 --- a/train.py +++ b/train.py @@ -393,7 +393,7 @@ if __name__ == '__main__': # Clip to limits keys = ['lr0', 'iou_t', 'momentum', 'weight_decay', 'hsv_s', 'hsv_v', 'translate', 'scale'] - limits = [(1e-4, 1e-2), (0.00, 0.70), (0.60, 0.95), (0, 0.001), (0, .8), (0, .8), (0, .8), (0, .8)] + limits = [(1e-4, 1e-2), (0.00, 0.70), (0.60, 0.99), (0, 0.001), (0, .8), (0, .8), (0, .8), (0, .8)] for k, v in zip(keys, limits): hyp[k] = np.clip(hyp[k], v[0], v[1])