Unfortunately i cannot share the code because it is proprietary. I have checked the inference accuracy of the two models converted with batch size = 1 which gives no warning and batch size = 8 the result is quite similar. I have checked the output pixel’s value in both cases. In the next days i will try to build a different version of the code with no proprietary restriction and if the error persists i will share the code.
Thanks.