Ошибка заключается в следующем:
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()