当前位置: 首页 > 工具软件 > Scalar > 使用案例 >

Expected object of scalar type Long but got scalar type Float

申屠泉
2023-12-01

背景:训练模型的时候报错Expected object of scalar type Long but got scalar type Float

解决:

方法一:
targets =torch.LongTensor(targets)
方法二:
targets = targets.to(self.device,dtype=torch.int64)

 

 类似资料:

相关阅读

相关文章

相关问答