فهرست منبع

fix Tensor.numpy()[0] to float(Tensor) to adapt 0D (#86)

Zhou Wei 2 سال پیش
والد
کامیت
a0f6163b86
1فایلهای تغییر یافته به همراه1 افزوده شده و 4 حذف شده
  1. 1 4
      paddlers/models/ppdet/modeling/tests/test_yolov3_loss.py

+ 1 - 4
paddlers/models/ppdet/modeling/tests/test_yolov3_loss.py

@@ -356,10 +356,7 @@ class TestYolov3LossOp(unittest.TestCase):
             x, t, gtbox, anchor, self.downsample_ratio, self.scale_x_y)
         for k in yolo_loss2:
             self.assertAlmostEqual(
-                yolo_loss1[k].numpy()[0],
-                yolo_loss2[k].numpy()[0],
-                delta=1e-2,
-                msg=k)
+                float(yolo_loss1[k]), float(yolo_loss2[k]), delta=1e-2, msg=k)
 
 
 class TestYolov3LossNoGTScore(TestYolov3LossOp):