I think, you almost solved the problem.
the above error says that either model or data is in GPU and the other is in CPU.
use .cuda() on both model and data.
1 Like
I think, you almost solved the problem.
the above error says that either model or data is in GPU and the other is in CPU.
use .cuda() on both model and data.