aft-pytorch icon indicating copy to clipboard operation
aft-pytorch copied to clipboard

can run on cpu but failed in gpu,why?

Open TangDL opened this issue 3 years ago • 1 comments

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() but it can't work, please help!

TangDL avatar Oct 12 '21 13:10 TangDL

Hey,

Thank you, this project is not actively maintained. Though, I'll check why this is happening.

Appreciate your patience!

rish-16 avatar Oct 15 '21 04:10 rish-16