IsmatseUPM

Hi, @HamzaFarooq013 Have you been able to solve that problem? I have kind of the same shape issue while training and I have no idea why.