I'm getting this error : indices should be either on cpu or on the same device as the indexed tensor (cpu) Even though I have specified a single gpu
I'm getting this error :
indices should be either on cpu or on the same device as the indexed tensor (cpu)
Even though I have specified a single gpu