报错如下:
Input type (torch.cuda.FloatTensor) :代表输入在 GPU 上
weight type (torch.FloatTensor):代表模型在 CPU 上
需要将模型转到 GPU 上,即:model.cuda()
mb62b19580f1ddc ©著作权
报错如下:
Input type (torch.cuda.FloatTensor) :代表输入在 GPU 上
weight type (torch.FloatTensor):代表模型在 CPU 上
需要将模型转到 GPU 上,即:model.cuda()
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M