99 Star 796 Fork 1.4K

MindSpore / models

 / 详情

加载预训练模型报错

REJECTED
Question
创建于  
2023-05-31 17:51

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类,有没有办法解决

评论 (3)

烈火燎原 创建了Documentation

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
恭喜你,你已经学会了使用命令来打标签,接下来就在下面的评论里打上标签吧!

烈火燎原 任务类型Documentation 修改为Question
烈火燎原 修改了描述
wuweikang 任务状态TODO 修改为ACCEPTED
wuweikang 负责人设置为wangzeyangyi
wuweikang 关联项目设置为MindSpore Issue Assistant
wuweikang 计划开始日期设置为2023-06-01
wuweikang 计划截止日期设置为2023-07-06
wuweikang 添加了
 
mindspore-assistant
标签
wuweikang 添加了
 
sig/modelzoo
标签
wangzeyangyi 负责人wangzeyangyi 修改为wuweikang
wuweikang 添加了
 
support-group
标签

该问题建议走需求流程,此issue关闭,谢谢!

wuweikang 任务状态ACCEPTED 修改为REJECTED

登录 后才可以发表评论

状态
负责人
项目
里程碑
Pull Requests
关联的 Pull Requests 被合并后可能会关闭此 issue
分支
开始日期   -   截止日期
-
置顶选项
优先级
预计工期 (小时)
参与者(3)
6573907 hw kk 1584444816
1
https://gitee.com/mindspore/models.git
git@gitee.com:mindspore/models.git
mindspore
models
models

搜索帮助