My code works fine on the CPU, but when I move everything to GPU and try to run it, I get an error on this line:
print(ctx["input"])
The shape of ctx["input"] is torch.Size([1, 50]) and its dtype is torch.float32.
The error says: “RuntimeError: CUDA error: invalid configuration argument”.
Actually, the error appears whenever I try to do anything with ctx["input"].
How can I fix this?
This is the entire traceback: dpaste/qDRq (Python) 3