环境paddleseg2.3  paddle2.2.1

使用fcn8s+vgg，已经训练完了，在测试时报错。

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, 2, 512, 512], X's size = 524288, 'shape' is [512, 512], the capacity of 'shape' is 262144.
[Hint: Expected capacity == in_size, but received capacity:262144 != in_size:524288.] (at ..\paddle/fluid/operators/reshape_op.cc:224)
[operator < reshape2 > error]

网上搜到的都是tensorflow，并且和我的情况不太像。求教如何改正。