PyTorch使用预训练模型-CFANZ编程社区

PyTorch使用预训练模型

妖妖妈

阅读 125

2023-02-22

state_dict()

功能：获取模型当前的参数，以一个有序字典形式返回。

这个有序字典中，key 是各层参数名，value 就是参数。

load_state_dict(state_dict)

功能：将 state_dict 中的参数加载到当前网络，常用于 finetune。

python怎么使用预训练的模型_PyTorch使用预训练模型

相关推荐
香小蕉
 pytorch加载预训练模型
香小蕉 79 0 0
慎壹
 torchvision pytorch预训练模型目标检测使用
慎壹 11 0 0
鱼满舱
 pytorch中torchvision读取预训练模型
鱼满舱 60 0 0
天悦哥
 PyTorch实现BERT预训练模型转化指南
天悦哥 31 0 0
知年_7740
 修改PyTorch预训练模型的下载路径
知年_7740 72 0 0
悲催博士僧
 bert预训练模型的使用
悲催博士僧 72 0 0
Python芸芸
 最强 NLP 预训练模型库 PyTorch-Transformers 正式开源：支持 6 个预训练框架，27 个预训练模型
Python芸芸 61 0 0
洒在心头的阳光
 Pytorch只更新预训练模型的部分参数
洒在心头的阳光 59 0 0
律楷粑粑
 预训练模型介绍
律楷粑粑 39 0 0
一叶随风_c94d
 pytorch官方实现加载预训练efficientnetv2模型
一叶随风_c94d 71 0 0

精彩评论（0）