pytorch保存模型方法_model.save_pretrained-程序员宅基地

技术标签: pytorch  

Pytorch 有两种保存模型的方式,都是通过调用pickle序列化方法实现的。

第一种方法只保存模型参数。第二种方法保存完整模型。推荐使用第一种,第二种方法可能在切换设备和目录的时候出现各种问题。

1.保存模型参数方法:

print(model.state_dict().keys())                                # 输出模型参数名称

# 保存模型参数到路径"./data/model_parameter.pkl"
torch.save(model.state_dict(), "./data/model_parameter.pkl")
new_model = Model()                                                    # 调用模型Model
new_model.load_state_dict(torch.load("./data/model_parameter.pkl"))    # 加载模型参数     
new_model.forward(input)                                               # 进行使用

2.保存完整模型(不推荐)

torch.save(model, './data/model.pkl')        # 保存整个模型
new_model = torch.load('./data/model.pkl')   # 加载模型

3.Transfomers库预训练模型的加载

# 使用transformers预训练后进行保存
model.save_pretrained(model_path)                              
tokenizer.save_pretrained(tokenizer_path)

# 预训练模型使用 `from_pretrained()` 重新加载
model.from_pretrained(model_path)                              
tokenizer.from_pretrained(tokenizer_path)

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/Obolicaca/article/details/123598171

智能推荐

狗血的Mybatis文件路径问题(org.apache.ibatis.binding.BindingException: Invalid bound statement (not found))_org.apache.ibatis.builder.cacherefresolver org.apa-程序员宅基地

文章浏览阅读1.3w次,点赞5次,收藏6次。@[toc](org.apache.ibatis.binding.BindingException: Invalid bound statement (not found))1、问题描述  今天学习Mybatis运行程序的时候不出意外,出现异常了。org.apache.ibatis.binding.BindingException: Invalid bound statement (not ..._org.apache.ibatis.builder.cacherefresolver org.apache.ibatis.parsing.xnode

xtu-成对的字符串(1037)_1037成对的字符串-程序员宅基地

文章浏览阅读200次。Description有些字符串,如果满足下面的性质,则称为成对的字符串:a. 所有的字符在字符串中出现偶数次b. 每一对相同的字符之间不会有出现奇数次的字符现在给你一些字符串,请判断这些字符串是否为成对的字符串。输入:第一行是一个整数K,表示有多少个测试用例,以后每行一个测试用例。每行为一个字符串(长度不超过1000个字符)。输出:每行输出一个测试用例的结果。如果是,输出Yes,否则输出No。思路维护一个栈,遇到相同元素出栈否则入栈,最后查看栈是否为空即可,注意c++没有一键清空栈的函_1037成对的字符串

【C语言程序设计】百钱买百鸡(百钱百鸡,百鸡问题)!_c语言编程我国古代的《张丘建算经》中有这样一道著名的百鸡问题:“鸡翁一,值-程序员宅基地

文章浏览阅读1w次,点赞10次,收藏65次。中国古代数学家张丘建在他的《算经》中提出了一个著名的“百钱买百鸡问题”,鸡翁一,值钱五,鸡母一,值钱三,鸡雏三,值钱一,百钱买百鸡,问翁、母、雏各几何?实现过程:使用 for 语句对 3 种鸡的数嫌在事先确定好的范围内进行穷举并判断,对满足条件的 3 种鸡的数量按指定格式输出,否则进行下次循环。程序代码如下:#include <stdio.h>int main(){ int cock,hen,chick; /*定义变童为基本整型*/ f..._c语言编程我国古代的《张丘建算经》中有这样一道著名的百鸡问题:“鸡翁一,值

WebBrowser 控件实现地址栏-程序员宅基地

文章浏览阅读261次。WebBrowser 控件具有多个与导航相关的属性、方法和事件。使用下面的成员可以将控件导航到特定 URL、在导航历史记录列表中向后和向前移动,还可以加载当前用户的主页和搜索页: Url Navigate GoBack GoForward GoHome ..._webbrowser控件 地址栏 工具条

移动端vue项目构建(二)配置环境 vue-router路由 VUX ui框架 axios请求等等~~-程序员宅基地

文章浏览阅读112次。首先说一下,此项目的配置环境插件,因为是移动端所以有一部分考虑的都是移动端 :路由vue-router 路由路由是vue项目中很重要的东西,构建的时候一路回车就是选好了。Ui框架 vuxUi框架集成的是VUXVux 基于WeUI和Vue(2.x)开发的移动端UI组件库,主要服务于微信页面官网:https://vux.li/#/ui库..._vue 开发移动端项目的 路由结构

ListView属性-程序员宅基地

文章浏览阅读210次。1. 背景色:listView设置背景色android:background="@drawable/bg",拖动或者点击list空白位置的时候发现ListItem都变成黑色。 因为默认的ListItem背景是透明的,而ListView的背景是固定不变的,所以在滚动条滚动的过程中如果实时地去将当前每个Item的显示内容跟背景进行混合运算,所以android系统为了优化这个过程用,就使用...

随便推点

网盘搜索引擎_tuoniao.me-程序员宅基地

文章浏览阅读9.9k次,点赞2次,收藏5次。推荐盘多多: http://www.panduoduo.netBdsola: http://www.3134.cc潜力股(保持更新与收录)探索云盘搜索: http://tansuo233.comBDY搜: http://www.bdysou.com盘窝窝: http://www.panww.com百度网盘搜索: http://uzi..._tuoniao.me

从键盘上输入5个数,输出最大、最小元素的值以及它们的下标-程序员宅基地

文章浏览阅读3.5k次。#defineN5#include<stdio.h>voidmain(){inti,j,k,max,min;staticinta[5];for(i=0;i<5;i++)scanf("%d",&a[i]);max=min=a[0];j=k=0;for(i=0;i<5;i++){if(max<a[..._从键盘输入5个数,输出最大、最小元素的值以及它们的下标。

BigPipe: Pipelining web pages for high performance-程序员宅基地

文章浏览阅读335次。Site speed is one of the most critical company goals for Facebook. In 2009, we successfully made Facebook site twice as fast, which was blogged in this post. Several key innovations from our enginee

超详细 kafka 入门(最佳实践)-程序员宅基地

文章浏览阅读297次。今日推荐6年,终拿腾讯 offer!都2021年了,你还在用Jenkins?赶快看看这些替代方案吧!盘点 12 个 GitHub 上的高仿项目CTO 说了,用错 @Autowire..._kafka实践

feign踩坑之报retryableRibbonLoadBalancingHttpClient_error creating bean with name 'retryableribbonload-程序员宅基地

文章浏览阅读5.3k次,点赞4次,收藏4次。我这边用的是springcloud2.1.0.RELEASE版本,spring-cloud-starter-openfeign用的默认版本。话不多说,直接上报错图Caused by: org.springframework.beans.factory.UnsatisfiedDependencyException: Error creating bean with name 'retryableRibbonLoadBalancingHttpClient' defined in org.spring._error creating bean with name 'retryableribbonloadbalancinghttpclient' defin

STM32CubeMX安装包 V6.0.1 + JAVA + V6.2.0_cubemx6.0.1下载-程序员宅基地

文章浏览阅读3.7k次,点赞7次,收藏7次。STM32CubeMX安装包 V6.0.1 + JAVA链接:https://pan.baidu.com/s/13bRpCvDzaCfi-I6Y3U37-g提取码:hqr1复制这段内容后打开百度网盘手机App,操作更方便哦_cubemx6.0.1下载