改变mobilenetv2网络后没办法使用预训练权重怎么半

改变mobilenetv2网络后没办法使用预训练权重怎么半,第1张

题主是否想询问“改变mobilenetv2网络后没办法使用预训练权重怎么办”?<br>1、首先对败逗洞察枯于微调后的网络模型可以将load_state_dict()函数中的参数strict=False。

2、指圆其次当strict = True时,要求加载的权重文件中的参数与模型中的完全匹配,设置strict = False时,即可使用权重文件中与模型匹配层的数据,其余的默认初始化。

3、最后当模型更改后不能如上述情况加载预训练权重时,可以考虑获取原模型的权重加载到新模型上。

左右

权重文件的产生,是根腊闷则据模型训练的过程而产生的。首先,根据训练数据,建立模型,定义模型的参数;然后,使用梯度轮棚下降法,不断调整参数,使模型的损失函数最小;最后,收敛到最优解,得到模型的权重参数,罩凳即权重文件。


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/tougao/12223892.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2023-05-22
下一篇 2023-05-22

发表评论

登录后才能评论

评论列表(0条)

保存