I am struggling with this tensor.cpu() problem 116
It seems like the latest pytorch upstream code had not solved this completely yet.
TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.