fix bias problem while training (PaddlePaddle#2161)

heavengate · Feb 2, 2021 · 524f56f · 524f56f
1 parent b83aab2
commit 524f56f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dygraph/ppdet/engine/trainer.py b/dygraph/ppdet/engine/trainer.py
@@ -115,9 +115,10 @@ def _init_metrics(self):
             return
         if self.cfg.metric == 'COCO':
             # TODO: bias should be unified
+            bias = 1 if 'bias' in self.cfg else 0
             self._metrics = [
                 COCOMetric(
-                    anno_file=self.dataset.get_anno(), bias=self.cfg.bias)
+                    anno_file=self.dataset.get_anno(), bias=bias)
             ]
         elif self.cfg.metric == 'VOC':
             self._metrics = [