ERROR in inference()

[rainley123]

First, when I use the inference() in resnet.py, assert will have an error, then I remove the assert, just give a assignment, and another error occurs, the shape of shortcut and x is not equal, they can not add together.
I don't know what happened

[rainley123]

And I also find that the structure of the pre-trained model is right, however it is nor equal to your code in inference()