lightweight_baseline_choi model shape issues

[AshishShukla-1992]

Hi ,

I tried my own trained model, but the output shape of my model is 1 x 21 x 3, but the model that you gave has an output shape of 1 x 672 x 32 x 32. I trained the model using https://github.com/SangbumChoi/MobileHumanPose repo. Could you please tell me why is this issue happening?

Thanks and Regards,
Ashish