Here is my new question:
I have tried by training over cpu, and overcome it. But now, I got this error:
“RuntimeError: Only tensors or tuples of tensors can be output from traced functions (getOutput at …\torch\csrc\jit\tracer.cpp:212)
(no backtrace available)”