PyTorch加载预训练模型实例(pretrained)
作者:Xie_learning 时间:2021-02-04 15:26:11
使用预训练模型的代码如下:
# 加载预训练模型
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)
来源:https://blog.csdn.net/Xie_learning/article/details/89176636
标签:PyTorch,预训练,模型,pretrained
0
投稿
猜你喜欢
MySQL5.7中的JSON基本操作指南
2024-01-27 20:38:34
网站通过W3C标准注意点
2010-01-14 20:15:00
php 方便水印和缩略图的图形类
2023-10-15 15:31:28
使用Rotate Master实现MySQL 多主复制的实现方法
2024-01-13 11:23:25
Yahoo!网站性能最佳体验的34条黄金守则—内容
2008-05-13 12:14:00
解决在pycharm运行代码,调用CMD窗口的命令运行显示乱码问题
2023-12-31 16:43:43
解决Win7 x64安装解压版mysql 5.7.18 winx64出现服务无法启动问题
2024-01-22 07:57:39
PHP PDOStatement::errorInfo讲解
2023-06-04 14:11:53
go语言学习之包和变量详解
2024-04-26 17:29:27
详解django中Template语言
2022-12-01 21:20:59
Python中的jquery PyQuery库使用小结
2023-05-27 11:08:15
基于Python记录一场2023的烟花
2022-01-08 19:57:07
mysql如何删除数据表和关联的数据表删除详情
2024-01-25 19:12:55
Python学习之循环方法详解
2021-04-13 21:07:15
Python字符串hashlib加密模块使用案例
2023-08-02 12:06:24
Python Pyperclip模块安装和使用详解
2023-06-25 13:52:11
配置MySQL与卸载MySQL实例操作
2024-01-19 21:55:26
美图秀秀web开放平台--PHP流式上传和表单上传示例分享
2023-11-15 08:34:42
PyTorch 如何将CIFAR100数据按类标归类保存
2023-01-10 06:01:03
彻底解决Python包下载慢问题
2021-07-17 16:02:27