C:\actions-runner\_work\pytorch\pytorch\builder\windows\pytorch\aten\src\ATen\native\cuda\Index
·
在跑代码的过程中，不知道为什么出现这个问题，而且也没有具体的报错位置，所以找了好长时间
问题在红框，实际的报错是因为分类的参数预先写的不对，参考别人的文章，也有类似的问题
可以检查一下自己写的分类的类别是否有变化
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
1
0- 0
已为社区贡献1条内容
所有评论(0)