TangDL

can run on cpu but failed in gpu,why?

1

RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for baddbmm) i set .cuda()...