未验证 提交 1d8b8107 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix test_architectures (#1147)

上级 34cc67c6
...@@ -19,6 +19,7 @@ from __future__ import print_function ...@@ -19,6 +19,7 @@ from __future__ import print_function
import unittest import unittest
import numpy as np import numpy as np
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
import os import os
import sys import sys
...@@ -70,6 +71,10 @@ class TestCascadeRCNN(TestFasterRCNN): ...@@ -70,6 +71,10 @@ class TestCascadeRCNN(TestFasterRCNN):
self.cfg_file = 'configs/cascade_rcnn_r50_fpn_1x.yml' self.cfg_file = 'configs/cascade_rcnn_r50_fpn_1x.yml'
@unittest.skipIf(
paddle.version.major < "2",
"Paddle 2.0 should be used for YOLOv3 takes scale_x_y as inputs, "
"disable this unittest for Paddle major version < 2")
class TestYolov3(TestFasterRCNN): class TestYolov3(TestFasterRCNN):
def set_config(self): def set_config(self):
self.cfg_file = 'configs/yolov3_darknet.yml' self.cfg_file = 'configs/yolov3_darknet.yml'
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册