成功解决RuntimeError: multi-target not supported at /pytorch/aten/src/THCUNN/generic/

计算交叉熵时报错：RuntimeError: multi-target not supported at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:15维度不匹配导致input, target的维度与F.cross_entropy()函数要求的维度不匹配解决方法：F.cross_entropy( input, target.sque

梦坠凡尘

6351人浏览 · 2022-03-17 18:19:49

梦坠凡尘 · 2022-03-17 18:19:49 发布

计算交叉熵时报错：
RuntimeError: multi-target not supported at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:15

维度不匹配导致
input, target的维度与F.cross_entropy()函数要求的维度不匹配

解决方法：

F.cross_entropy( input, target.squeeze() )

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

大模型赋能：模型字段别名的智能生成之旅

这不仅提高了数据分析的效率，还降低了用户使用数据分析平台的门槛，使得更多的业务人员能够参与到数据分析中来，为企业的决策提供更广泛的支持。而且，大模型生成的别名具有高度的一致性，避免了人工生成别名时可能出现的不一致问题，确保了数据仓库中数据的规范性和统一性。这不仅可以大大提高别名生成的效率，减轻数据开发人员的工作负担，还能有效提升别名的一致性和准确性，为数据治理工作注入新的活力。大模型技术的出现，为