AttributeError: module ‘mindspore.nn‘ has no attribute ‘CrossEntropyLoss‘
·
【报错】使用华为云 ModelArts 下 Notebook 进行 Mindspore 框架开发时遇到 AttributeError: module 'mindspore.nn' has no attribute 'CrossEntropyLoss' 报错。
【原因】新版 Notebook 采用的 mindspore 将 CrossEntropyLoss 函数迁移至 nn.transformer 下。
【解决办法】使用 nn.transformer.CrossEntropyLoss() 即可。
修改前：
loss_fn = nn.CrossEntropyLoss()
修改后：
loss_fn = nn.transformer.CrossEntropyLoss()
也可以切换至旧版 Notebook 进行开发 。（试了一下，并不可以）
更多可参考 Mindspore官方文档。
昇腾计算产业是基于昇腾系列（HUAWEI Ascend）处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务，https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)