pse_predict函数运行报错ValueError: Dimension 0 in both shapes must be equal

[Adhders]

Dimension 0 in both shapes must be equal, but are 1 and 512. Shapes are [1,1,128,512] and [512,256,1,1]. for 'Assign_78' (op: 'Assign') with input shapes: [1,1,128,512], [512,256,1,1]