device mismatch issue #1
Description
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
{{ message }}
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