未验证 提交 233b3641 编写于 作者: L LokeZhou 提交者: GitHub

Support 0 dimension tensor modification (#7250)

上级 3deb2c10
...@@ -356,10 +356,7 @@ class TestYolov3LossOp(unittest.TestCase): ...@@ -356,10 +356,7 @@ class TestYolov3LossOp(unittest.TestCase):
x, t, gtbox, anchor, self.downsample_ratio, self.scale_x_y) x, t, gtbox, anchor, self.downsample_ratio, self.scale_x_y)
for k in yolo_loss2: for k in yolo_loss2:
self.assertAlmostEqual( self.assertAlmostEqual(
yolo_loss1[k].numpy()[0], float(yolo_loss1[k]), float(yolo_loss2[k]), delta=1e-2, msg=k)
yolo_loss2[k].numpy()[0],
delta=1e-2,
msg=k)
class TestYolov3LossNoGTScore(TestYolov3LossOp): class TestYolov3LossNoGTScore(TestYolov3LossOp):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册