RuntimeError: For 'load_param_into_net', rcnn_cls.fpn_cls.cls_scores.weight in the argument 'net' should have the same shape as rcnn_cls.fpn_cls.cls_scores.weight in the argument 'parameter_dict'. But got its shape (1024, 3) in the argument 'net' and shape (1024, 81) in the argument 'parameter_dict'.May you need to check whether the checkpoint you loaded is correct or the batch size and so on in the 'net' and 'parameter_dict' are same.
环境mindspore1.8.0-cann5.1.2-py3.7-euler2.8.3
代码https://gitee.com/mindspore/models/tree/r1.8/official/cv/maskrcnn
能不能修改最后一层为3,客户又不是一定是81类,有没有办法解决
Please assign maintainer to check this issue.
请为此issue分配处理人。
@fangwenyi @chengxiaoli @wuweikang
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。
Please add labels (comp or sig), also you can visit https://gitee.com/mindspore/community/blob/master/sigs/dx/docs/labels.md to find more.
为了让代码尽快被审核,请您为Pull Request打上 组件(comp)或兴趣组(sig) 标签,打上标签的PR可直接推送给责任人进行审核。
更多的标签可以查看https://gitee.com/mindspore/community/blob/master/sigs/dx/docs/labels.md
以组件相关代码提交为例,如果你提交的是data组件代码,你可以这样评论:
//comp/data
当然你也可以邀请data SIG组来审核代码,可以这样写:
//sig/data
另外你还可以给这个PR标记类型,例如是bugfix或者是特性需求:
//kind/bug or //kind/feature
恭喜你,你已经学会了使用命令来打标签,接下来就在下面的评论里打上标签吧!
该问题建议走需求流程,此issue关闭,谢谢!
登录 后才可以发表评论