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 to your model, so it’s still unclear to me, how that many arguments are being passed.