当前位置：网站首页>MindSpore：损失函数问题

MindSpore：损失函数问题

2022-08-04 09:03:00 【小乐快乐】

问题描述：

【功能模块】

【操作步骤&问题现象】

1、mindspore中有和pytoch功能一模一样的binary_cross_entropy_with_logits的loss函数吗

2、

【截图信息】

【日志信息】（可选，上传日志内容或者附件）

解决方案：

nn.BCEWithLogitsLoss和ops.BCEWithLogitsLoss 这两个 GPU和Ascend 都不支持，没有这两个

通过查看MindSpore官网API，对上面两个loss是支持Ascend硬件平台的。

·https://www.mindspore.cn/doc/api_python/zh-CN/r1.2/mindspore/ops/mindspore.ops.BCEWithLogitsLoss.html#mindspore.ops.BCEWithLogitsLoss

·https://www.mindspore.cn/doc/api_python/zh-CN/r1.2/mindspore/nn/mindspore.nn.BCEWithLogitsLoss.html#mindspore.nn.BCEWithLogitsLoss

如果使用r1.1以及之前的版本是不能直接使用的，但是底层算子应该都是不缺的，你可以试试直接copy https://gitee.com/mindspore/mindspore/blob/master/mindspore/nn/loss/loss.py#L924 这个Cell

版权声明
本文为[小乐快乐]所创，转载请带上原文链接，感谢
https://blog.csdn.net/weixin_45666880/article/details/126149146