68-Retinaface-Pytorch-version icon indicating copy to clipboard operation
68-Retinaface-Pytorch-version copied to clipboard

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

Open Happythinker opened this issue 3 years ago • 2 comments

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

Happythinker avatar Sep 27 '21 09:09 Happythinker

i met the same question,how to solve it

876399730 avatar Feb 18 '22 15:02 876399730

image_batch.cuda() 强制作为模型的输入

cucdengjunli avatar Sep 06 '23 20:09 cucdengjunli