Oh, very sorry for that, I have not tried your model yet.
As mentioned above, could you try to run standalone script as well?
See Inferring resnet18 classification etlt model with python - #9 by Morganh
and Inferring resnet18 classification etlt model with python - #40 by Morganh