PyTorch加载预训练模型实例(pretrained)
作者：Xie_learning 发布时间：2021-02-04 15:26:11
标签：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)
来源：https://blog.csdn.net/Xie_learning/article/details/89176636
0
投稿猜你喜欢
FME如何使用Python？安装FME后，在FME的安装路径中，找到 fmeobjects 文件夹（C:\Program Files\FME- docutils 的官方工具地址为：https://docutils.sourceforge.io/目前的更新主要是在版本和使用手册的更新上
- 在SQL查询中，关键词Like可提供模糊查询功能，它通常与通配符一起使用。1 Like条件适用数据库字段类型 &nbs
- Date.prototype.format = function(format){ var o = { "M+" : t
- 1. go的依赖管理发展GOPATH 所有的依赖都放置在同一路径下，可以复用下载的依赖，但是当不同项目需要依赖不同版本的依赖时，就很容易出现
- 本文实例讲述了Python3对称加密算法AES、DES3。分享给大家供大家参考，具体如下：python3.6此库安装方式，需要pip3 in
- 条件选取：torch.where(condition, x, y) → Tensor返回从 x 或 y 中选择元素的张量，取决于 condi
- 1、你需要通过指定的文本模式去检查字符串的开头或者结尾，比如文件名后缀，URL Scheme 等等。检 查 字 符 串 开 头 或 结 尾
- 昨天，一同事发过来的一道数据库题目，就是哪种经典的父子级 ID 在同一数据库表中设计类型。需要在原表中添加一个字段，同时，将该节点的父子级详
- 本文实例为大家分享了Python时间戳使用和相互转换的具体代码，供大家参考，具体内容如下1.将字符串的时间转换为时间戳方法: &n
- 什么是 PycharmPycharm 是目前最好用的 Python 编辑器，自带文本高亮、版本管理、数据库连接、断点调试、虚拟环境和包管理的
- Numpy中提供了concatenate,append, stack类(包括hsatck、vstack、dstack、row_stack、c
- 开发背景：每次项目都要写数据库、烦死了。。然后就每次数据库都要花很多时间。烦死了！不如写个通用的增删查改，以不变应万变！特性：搭建通用增删查
- pycharm安装lxml库失败怎么办？众所周知，在pycharm中安装lxml库，一般的安装流程流程为file->settings-
- SQLserver代理已经启动了，服务里的SQLServerAgent响应服务也已经开启了 但是启动作业的时候还是提示“错误22022：SQ
- 在golang中，网络协议已经被封装的非常完好了，想要写一个Socket的Server，我们并不用像其他语言那样需要为socket、bind
- 使用os.environ来读取和修改环境变量：import osprint (os.environ["TEMP"])my
- 一.GO程序目录结构在GOPATH目录下的结构--bin(存放编译后生成的可执行文)|----hello.exe（可执行文件）--pkg(存
- 模块概述如果说模块是按照逻辑来组织 Python 代码的方法, 那么文件便是物理层上组织模块的方法。 因此, **一个文件被看作是一个独立模
- type()函数：使用type()函数可以判断对象的类型，如果一个变量指向了函数或类，也可以用type判断。如：class Student(
