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

[11167 (Hong)]

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’

[albanD]

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

[11167 (Hong)]

thank you. I known the reason for this issue.