aft-pytorch
aft-pytorch copied to clipboard
can run on cpu but failed in gpu,why?
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!
Hey,
Thank you, this project is not actively maintained. Though, I'll check why this is happening.
Appreciate your patience!