When I connect to mlmodel with my swift code it has generate following error

[corem1] Output backing for Identity is not compatiple with the model's output feature description.

[coremi] Failure in -bindOutputBuffers: outputBackings: error:. Image size 1 x 512 not in allowed set of image sizes.

Following display to spec.discription on python env

input { name: "input_2" type { imageType { width: 512 height: 768 colorSpace: RGB imageSizeRange { widthRange { lowerBound: 512 upperBound: 512 } heightRange { lowerBound: 768 upperBound: 768 } } } } } output { name: "Identity" type { imageType { width: 512 height: 768 colorSpace: GRAYSCALE } } }

Please help to solve above problem

When I connect to mlmodel with my swift code it has generate following error
 
 
Q