68-Retinaface-Pytorch-version
68-Retinaface-Pytorch-version copied to clipboard
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
i met the same question,how to solve it
image_batch.cuda() 强制作为模型的输入