五、RuntimeError: Тип ввода (torch.cuda.FloatTensor) и тип веса (torch.FloatTensor)

Ошибка заключается в следующем:

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

Тип ввода (torch.cuda.FloatTensor): представляет вход на GPU
Тип веса (torch.FloatTensor): представляет модель на CPU Модель
нужно передать на GPU, а именно: model.cuda()

Guess you like

Origin blog.csdn.net/panchang199266/article/details/128153179