[问题求助]
模型需要使用两个loss_fn时,应该怎么操作?
【功能模块】
复现MTCNN网络模型,对输出的class、box分别计算loss
【操作步骤&问题现象】
1、此处是否可以使用WithLossCell&TrainOneStepCell方法呢?如果可以应该怎样操作呢?
2、如果不可以用,是否有其他方法可以实现,能否提供参考样例?
在此一键设置昵称,即可参与社区互动!
发表于2021年02月20日 16:14:56
直达本楼层的链接
板凳
显示全部楼层
一个epoch结束后报错,请问如何解决?
评论
您好,已经联系技术支撑帮助看,会尽快给您答复。 ... 查看全部您好,看报错可能是数据读取的问题,方便的话可以展示下脚本,我们进行定位分析,谢谢。 ... 查看全部 |
|
已参考其他官方模型自定义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
... 查看全部