背景:训练模型的时候报错Expected object of scalar type Long but got scalar type Float
解决:
方法一:
targets =torch.LongTensor(targets)
方法二:
targets = targets.to(self.device,dtype=torch.int64)
本文解决了一个常见的PyTorch模型训练错误,即期望长整型目标变量但收到浮点型。提供了两种解决方案,一种是使用torch.LongTensor转换目标变量,另一种是通过指定设备和数据类型为int64来转换。
背景:训练模型的时候报错Expected object of scalar type Long but got scalar type Float
解决:
方法一:
targets =torch.LongTensor(targets)
方法二:
targets = targets.to(self.device,dtype=torch.int64)
1万+
240
5万+
5万+