X2Paddle
X2Paddle copied to clipboard
x2paddle转换yolov5后推理报错
在将pytorch的yolov5转换到paddle过程中eval报错
-
错误信息 ValueError: (InvalidArgument) The 'shape' in ReshapeOp is invalid. The input tensor X'size must be equal to the capacity of 'shape'. But received X's shape = [1, 512, 1, 1], X's size = 512, 'shape' is [2, 512], the capacity of 'shape' is 1024. [Hint: Expected capacity == in_size, but received capacity:1024 != in_size:512.] (at ../paddle/fluid/operators/reshape_op.cc:234) [operator < reshape2 > error]
-
错误截图
具体信息
-
转换模型后用处
- [ ] 使用 Paddle-Lite 做移动端推理