当前位置:网站首页>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
边栏推荐
猜你喜欢

Post-94 Byte P7 posted the salary slip: It's really good to make up for this...

TiFlash 源码阅读(五) DeltaTree 存储引擎设计及实现分析 - Part 2

telnet远程登录aaa模式详解【华为eNSP】

sql在字段重复时 对某个字段根据最新时间取数

TiCDC同步延迟问题处理

Recommend several methods that can directly translate PDF English documents

技术实现 | 图像检索及其在淘宝的应用

BFM模型和Landmarks可视化

RL学习笔记(一)

leetcode二叉树系列(二叉搜索树篇)
随机推荐
LVGL的多语言转换工具--字体设置的好助手
VRRP + MSTP configuration, huawei eNSP experiment 】 【
Shared_preload_libraries导致很多语法不支持
Detailed explanation of telnet remote login aaa mode [Huawei eNSP]
请你谈谈网站是如何进行访问的?【web领域面试题】
sync-diff-inspector 使用实践
Libpq 是否支持读写分离配置
从底层看 Redis 的五种数据类型
抬升市场投资情绪,若羽臣是否还需“自身硬”?
记录十条工作中便利的API小技巧
外包干了四年,秋招终于上岸了
Thread类的基本使用。
字符串与正则表达式(C#)
VRRP+MSTP配置详解【华为eNSP实验】
【高并发基石】多线程、守护线程、线程安全、线程同步、互斥锁
路由/三层交换机DHCP下发地址详解【华为eNSP】
发现WRH几个表被锁了,怎么办?
About Oracle RAC 11g rebuilding the disk group
layout manager
[Punctuality Atom STM32 Serial] Chapter 2 STM32 Introduction Excerpted from [Punctual Atom] MiniPro STM32H750 Development Guide_V1.1