PyTorch加载预训练模型实例(pretrained)
更新时间：2020年01月17日 08:46:14 作者：Xie_learning
今天小编就为大家分享一篇PyTorch加载预训练模型实例(pretrained)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
使用预训练模型的代码如下：
# 加载预训练模型
resNet50 = models.resnet50(pretrained=True)
ResNet50 = ResNet(Bottleneck, [3, 4, 6, 3], num_classes=2)
# 读取参数
pretrained_dict = resNet50.state_dict()
model_dict = ResNet50.state_dict()
# 将pretained_dict里不属于model_dict的键剔除掉
pretrained_dict = {k: v for k, v in pretrained_dict.items() if k in model_dict}
# 更新现有的model_dict
model_dict.update(pretrained_dict)
# 加载真正需要的state_dict
ResNet50.load_state_dict(model_dict)
以上这篇PyTorch加载预训练模型实例(pretrained)就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家详细介绍了如何利用Pandas实现对数据进行移动计算，文中的示例代码讲解详细，对我们了解Pandas有一定帮助，需要的可以参考一下2022-07-07
下面小编就为大家带来一篇Python实现好友全头像的拼接实例(推荐)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-06-06
这篇文章主要介绍了Python入门教程(十九)python的函数,函数是组织好的，可重复使用的，用来实现单一，或相关联功能的代码段,需要的朋友可以参考下2023-04-04
今天小编就为大家分享一篇python 画函数曲线示例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
今天小编就为大家分享一篇python交互界面的退出方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-02-02
这篇文章主要介绍了Python使用Flask框架同时上传多个文件的方法,实例分析了Python中Flask框架操作文件实现上传的技巧,需要的朋友可以参考下2015-03-03
今天小编就为大家分享一篇python 同时运行多个程序的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
这篇文章主要介绍了python爬虫爬取bilibili网页基本内容，用爬虫爬取bilibili网站排行榜游戏类的所有名称及链接，下面来看看具体的实现过程吧,需要的朋友可以参考一下2022-01-01
这篇文章主要介绍了在Python中使用PIL模块对图片进行高斯模糊处理的教程,这个无图形界面的脚本代码非常简单,需要的朋友可以参考下2015-05-05
Python使用selenium实现网页用户名 密码 验证码自动登录功能
这篇文章主要介绍了Python使用selenium实现网页用户名 密码 验证码自动登录功能，实现思路很简单，感兴趣的朋友跟随脚本之家小编一起学习吧2018-05-05
最新评论