From bed4da5dbb2ed16b83f5e9f18fd619dd873e1e84 Mon Sep 17 00:00:00 2001 From: Jirka Date: Tue, 13 Oct 2020 22:03:48 +0200 Subject: [PATCH] comment missing arg --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 80d9eec1e823..fb085031392a 100644 --- a/train.py +++ b/train.py @@ -471,7 +471,7 @@ def train(hyp, opt, device, tb_writer=None): 'obj_pw': (1, 0.5, 2.0), # obj BCELoss positive_weight 'iou_t': (0, 0.1, 0.7), # IoU training threshold 'anchor_t': (1, 2.0, 8.0), # anchor-multiple threshold - 'anchors': (2, 2.0, 10.0), # anchors per output grid (0 to ignore) + #'anchors': (2, 2.0, 10.0), # anchors per output grid (0 to ignore) 'fl_gamma': (0, 0.0, 2.0), # focal loss gamma (efficientDet default gamma=1.5) 'hsv_h': (1, 0.0, 0.1), # image HSV-Hue augmentation (fraction) 'hsv_s': (1, 0.0, 0.9), # image HSV-Saturation augmentation (fraction)