只想努力赚钱的小小小M 2021-12-22 21:56 采纳率: 0%
浏览 69

mobileNetV2代码理解,冻结参数

mobileNetV2网络搭建出现的问题,为什么要冻结参数,以及
pre_dict={k:v for k,v in pre_weights.items() if net.state_dict()[k].numel()==v.numel()}
的理解不太明白

img

  • 写回答

1条回答 默认 最新

  • 关注

    pre_dict={k:v for k,v in pre_weights.items() if net.state_dict()[k].numel()==v.numel()}

    是从 pre_weights字典中筛选出它的key与net.state_dict()返回的字典对应的key的对象.numel()等于v.numel()的项目

    评论

报告相同问题?

问题事件

  • 创建了问题 12月22日