模型需要使用两个loss_fn时，应该怎么操作？_MindSpore_昇腾论坛

标签

您还可以添加5个标签

没有搜索到和“关键字”相关的标签

云产品
解决方案
技术领域
通用技术
平台功能

知识搬运工发帖: 5粉丝: 1 发消息 + 关注	发表于2021年02月20日 09:59:14 阅读97 回复3 直达本楼层的链接楼主阅读模式显示全部楼层 [问题求助] 模型需要使用两个loss_fn时，应该怎么操作？【功能模块】复现MTCNN网络模型，对输出的class、box分别计算loss 【操作步骤&问题现象】 1、此处是否可以使用WithLossCell&TrainOneStepCell方法呢？如果可以应该怎样操作呢？ 2、如果不可以用，是否有其他方法可以实现，能否提供参考样例？

	举报分享分享文章到朋友圈分享文章到微博复制链接到剪贴板

chengxiaoli

发帖: 135粉丝: 24

发消息 + 关注

发表于2021年02月20日 10:28:40

直达本楼层的链接

沙发显示全部楼层

您好，欢迎使用MindSpore

问题已经收到了，正在定位分析，会尽快给您回复。

慢慢学 2021-2-20 16:11 评论

已参考其他官方模型自定义WithLossCell进行训练，但训练一个epoch结束会报错如下内容，请问该怎么解决. [ERROR] ME(27588:11752,MainProcess):2021-02-20-16:06:34.804.560 [mindspore\dataset\engine\iterators.py:172] Got runtime err: Thread ID 1 Unexpected error. EOF buffer encountered. Users try to fetch data beyond the specified number of epochs. Line of code : 238 File : D:\jenkins\agent-working-dir\workspace\Compile_CPU_Windows\mindspore\mindspore\ccsrc\minddata\dataset\engine\tree_adapter.cc

... 查看全部

点赞评论引用举报

慢慢学

发帖: 5粉丝: 1

发消息 + 关注

发表于2021年02月20日 16:14:56

直达本楼层的链接

板凳显示全部楼层

一个epoch结束后报错，请问如何解决？

chengxiaoli 2021-2-20 16:15 评论

您好，已经联系技术支撑帮助看，会尽快给您答复。

... 查看全部

chengxiaoli 2021-2-20 16:56 评论

您好，看报错可能是数据读取的问题，方便的话可以展示下脚本，我们进行定位分析，谢谢。

... 查看全部

点赞评论引用举报

慢慢学发帖: 5粉丝: 1 发消息 + 关注	发表于2021年02月20日 17:19:38 直达本楼层的链接地板显示全部楼层已解决，感谢！原因：如图所示，num_epochs参数之前设置的是1，改成max_epoch就好了评论 chengxiaoli 2021-2-20 17:49 评论好的，谢谢您解决方法的分享 ... 查看全部
	点赞评论引用举报

设置昵称

选择版块