I could check the one piece of data whether model trained well

I’m trying to check if the result is good with one piece of data, but I keep seeing an error.
What should I do?


It occure the error

‘RuntimeError: CHECK_ARG(tensor->nDimension >= 2 && tensor->nDimension <= 5) failed at torch/csrc/cudnn/BatchNorm.cpp:13’

Isn’t your input x missing the batch dimension? Adding an unsqueeze(0) might help.

thank you. I known the reason for this issue.