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

Pytorch使用过程中 torchsummary.summary 中出现了上述错误,torchsummary是应用在pytorch中的一种结构表达方式。 if name main : model fishnet torchsummary.summary model.cuda , , , if name main : model fishnet torchsummary.summary model ...

2020-10-19 20:03 0 561 推荐指数:

查看详情

torch.cuda.FloatTensor

Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型,一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。 一般系统默认是torch.FloatTensor类型。例如data = torch.Tensor(2,3)是一个2*3的张量,类型 ...

Tue Dec 22 16:56:00 CST 2020 0 1141
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM