未验证 提交 99ad5a84 编写于 作者: W wangguanzhong 提交者: GitHub

fix_eval, test=dygraph (#2082)

上级 ac358bfe
...@@ -97,8 +97,8 @@ class Trainer(object): ...@@ -97,8 +97,8 @@ class Trainer(object):
def _init_metrics(self): def _init_metrics(self):
if self.mode == 'eval': if self.mode == 'eval':
if self.cfg.metric == 'COCO': if self.cfg.metric == 'COCO':
mask_resolution = self.model.mask_post_process.mask_resolution if hasattr( mask_resolution = self.model.mask_post_process.mask_resolution if getattr(
self.model, 'mask_post_process') else None self.model, 'mask_post_process', None) else None
self._metrics = [ self._metrics = [
COCOMetric( COCOMetric(
anno_file=self.dataset.get_anno(), anno_file=self.dataset.get_anno(),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册