Focal-Loss-Pytorch icon indicating copy to clipboard operation
Focal-Loss-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon yatengLG

Expected object of backend CPU but got backend CUDA for argument #3 'index'

Open wlsrick opened this issue 1 year ago • 0 comments

I've got this problem in line 51 alpha = self.alpha.gather(0,labels.view(-1))

How can I solve it? thanks a lot.

wlsrick avatar Jun 20 '23 09:06 wlsrick