Skip to content

Commit 7bc8241

Browse files
committed
add attributes for unittest
1 parent fb03dae commit 7bc8241

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python/paddle/fluid/tests/unittests/test_yolo_box_op.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -265,6 +265,8 @@ def initTestCase(self):
265265
self.x_shape = (self.batch_size, an_num * (5 + self.class_num), 13, 9)
266266
self.imgsize_shape = (self.batch_size, 2)
267267
self.scale_x_y = 1.
268+
self.iou_aware = False
269+
self.iou_aware_factor = 0.5
268270

269271

270272
if __name__ == "__main__":

0 commit comments

Comments
 (0)