Solved by ptrblck in post #4 
              
              
                To fix your new error message, it seems you would need to add the batch dimension to your input, as currently you are passing it as [C, H, W]. 
Try something like data = data.unsqueeze(0) before passing it to the model. 
The code you’ve posted does not show the code snippet where you pass the input …