From 2615ca37c33a916d9f8b546e2d99fc255196d32b Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Tue, 2 Feb 2021 15:36:58 +0800
Subject: [PATCH] fix bias problem while training (#2162)

---
 dygraph/ppdet/engine/trainer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/dygraph/ppdet/engine/trainer.py b/dygraph/ppdet/engine/trainer.py
index d18492a4e..771866f65 100644
--- a/dygraph/ppdet/engine/trainer.py
+++ b/dygraph/ppdet/engine/trainer.py
@@ -115,9 +115,10 @@ class Trainer(object):
             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 = [
-- 
GitLab