AttributeError: module ‘mindspore.nn‘ has no attribute ‘CrossEntropyLoss‘

【报错】使用华为云ModelArts下Notebook进行Mindspore框架开发时遇到 AttributeError: module 'mindspore.nn' has no attribute 'CrossEntropyLoss' 报错。

AttributeError: module ‘mindspore.nn‘ has no attribute ‘CrossEntropyLoss‘_第1张图片

【原因】新版 Notebook 采用的 mindspore 将 CrossEntropyLoss 函数迁移至 nn.transformer 下。

【解决办法】使用 nn.transformer.CrossEntropyLoss() 即可。

修改前:

loss_fn = nn.CrossEntropyLoss()

修改后:

loss_fn = nn.transformer.CrossEntropyLoss()

也可以切换至旧版 Notebook 进行开发 。(试了一下,并不可以)

AttributeError: module ‘mindspore.nn‘ has no attribute ‘CrossEntropyLoss‘_第2张图片

更多可参考 Mindspore官方文档。

你可能感兴趣的:(编程报错,python)