Expected object of scalar type Long but got scalar type Float

coding_zhang

9554人浏览 · 2019-04-28 18:33:56

coding_zhang · 2019-04-28 18:33:56 发布

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

解决：

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

HarmonyOS开发者社区

讨论HarmonyOS开发技术，专注于API与组件、DevEco Studio、测试、元服务和应用上架分发等。

更多推荐

对话AI、玩转开发，HarmonyOS开发助手挑战活动

本次活动分为初阶活动和高阶活动两部分，面向所有开发者进行招募，只要您想要尝试鸿蒙应用/元服务开发、鸿蒙应用/元服务适配手机、平板、折叠屏等多种设备、三方库鸿蒙化适配，均可以报名。基准：您的应用/元服务使用AI插件任一种/多种特性（元服务生成、元服务转换、三方库鸿蒙化、多设备适配），并上架或更新该元服务/应用。内容贡献 (25%)：在官方社区以及三方社区分享体验文章（完整性40%、逻辑性20%、阅读