fix test_detection. test=develop

revert-16190-refine_parallel_executor
dengkaipeng 6 years ago
parent abb5a9c726
commit 00e822d2a0

@ -484,7 +484,7 @@ class TestYoloDetection(unittest.TestCase):
program = Program()
with program_guard(program):
x = layers.data(name='x', shape=[30, 7, 7], dtype='float32')
img_size = layers.data(name='x', shape=[2], dtype='int32')
img_size = layers.data(name='img_size', shape=[2], dtype='int32')
boxes, scores = layers.yolo_box(x, img_size, [10, 13, 30, 13], 10,
0.01, 32)
self.assertIsNotNone(boxes)

Loading…
Cancel
Save