From 8e239684f8fb427533cb1225a1c552fc71698ea4 Mon Sep 17 00:00:00 2001 From: dengkaipeng Date: Sun, 13 Sep 2020 12:57:06 +0000 Subject: [PATCH] add comment --- yolov3/modeling.py | 1 + 1 file changed, 1 insertion(+) diff --git a/yolov3/modeling.py b/yolov3/modeling.py index 9cba02e..6bd9b9f 100644 --- a/yolov3/modeling.py +++ b/yolov3/modeling.py @@ -272,6 +272,7 @@ class YoloLoss(fluid.dygraph.Layer): downsample = 32 losses = [] + # YOLOv3 output fields is different between 'train' and 'eval' mode if len(inputs) == 6: output1, output2, output3, gt_box, gt_label, gt_score = inputs elif len(inputs) == 8: -- GitLab