当前位置：首页 > news >正文

深度学习Pytorch中的模型保存与加载方法

news 来源：原创 2025/5/17 17:52:32

深度学习:Pytorch中的模型保存与加载方法

在 PyTorch 中，模型的保存和加载对于模型的持久化和后续应用至关重要。这里详细介绍了两种主要方法：保存整个模型（包括架构和参数）和仅保存模型的状态字典。以下内容进一步完善了加载模型时使用 load_state_dict 方法的细节和参数设置。

1. 保存和加载整个模型

方法描述

这种方法涉及使用 torch.save 函数直接保存模型的整个对象，并通过 torch.load 来恢复整个模型，包括其架构和所有参数。

参数和设置

保存时的参数和设置：
- model: 需要保存的模型实例。
- filepath: 模型保存的文件路径，如 'vgg16_complete_model.pth'。
加载时的参数和设置：
- filepath: 保存的模型文件路径，用于加载模型。

示例代码

import torch
import torchvision.models as models# 初始化模型
model = models.vgg16(pretrained=True)# 保存整个模型
torch.save(model, 'vgg16_complete_model.pth')# 加载整个模型
loaded_model = torch.load('vgg16_complete_model.pth')
print("Loaded Model:", loaded_model)

优点

快速便捷，可以直接恢复整个模型，包括其架构和参数。

缺点

安全风险较高，因为使用了 pickle 序列化，可能在加载时执行未知代码。
文件体积较大，包含了整个模型的详细信息。

2. 仅保存和加载模型的状态字典

方法描述

此方法仅保存模型的参数，使用 state_dict 获取所有参数的字典，并通过先定义模型架构后使用 load_state_dict 方法加载参数。

参数和设置

保存时的参数和设置：
- model.state_dict(): 调用模型的 state_dict() 方法获取所有参数的字典。
- filepath: 参数字典保存的文件路径，如 'vgg16_model_state_dict.pth'。
加载时的参数和设置：
- filepath: 保存的状态字典文件路径，用于加载参数。
- 需要先重新定义模型结构以匹配状态字典。
- 使用 load_state_dict 方法将参数加载到模型。

示例代码

import torch
import torchvision.models as models# 初始化预训练模型
model = models.vgg16(pretrained=True)# 保存模型的状态字典
torch.save(model.state_dict(), 'vgg16_model_state_dict.pth')# 重新定义模型架构
model = models.vgg16()  # 重新初始化模型# 加载模型参数
model.load_state_dict(torch.load('vgg16_model_state_dict.pth'))
print("Loaded State Dict:", model)

`load_state_dict` 方法详解

load_state_dict 是一个 PyTorch 方法，用于将保存的状态字典加载到模型的参数中。
此方法确保所有保存的参数能正确映射到模型的相应参数上。
如果状态字典中的参数与模型架构不匹配，将抛出错误。

优点

安全性高，不涉及 pickle，避免潜在的代码执行风险。
文件体积小，因为仅包含参数。

缺点

需要在加载参数前重新定义模型架构，可能在结构复杂或难以重建时带来不便。

比较异同

相同点：
- 使用 torch.save() 和 torch.load() 函数来保存和加载。
- 都能有效地保存和恢复模型数据，方便模型的迁移和再使用。
不同点：
- 安全性：保存整个模型可能面临执行恶意代码的风险，而保存状态字典则相对安全。
- 便利性：保存整个模型可以直接加载使用，适合快速部署；而保存状态字典需要先定义架构，但提供更高的灵活性和安全性。
- 文件大小：整个模型文件通常较大；状态字典文件体积较小，仅包含参数。

这些详细的说明帮助开发者根据具体需求选择合适的方法来保存和加载 PyTorch 模型，确保模型的有效部署和应用，同时最大化效率和安全性。

深度学习Pytorch中的模型保存与加载方法

深度学习:Pytorch中的模型保存与加载方法在 PyTorch 中，模型的保存和加载对于模型的持久化和后续应用至关重要。这里详细介绍了两种主要方法：保存整个模型（包括架构和参数）和仅保存模型的状态字典。以下内容进一步完善了加载模型…...

编程日记 2025/5/17 17:52:32

(vue)启动项目报错The project seems to require pnpm but it‘s not installed

(vue)启动项目报错The project seems to require pnpm but it’s not installed 原因该错误信息表明你的项目需要使用 pnpm 作为包管理工具，但系统中尚未安装 pnpm。解决方法【1】删除pnpm.lock 【2】npm install -g pnpm 之后再重新启动 yarn报错&#xff0…...

编程日记 2025/5/17 7:55:33

【Leetcode 每日一题】3250. 单调数组对的数目 I

问题背景给你一个长度为 n n n 的正整数数组 n u m s nums nums。如果两个非负整数数组 ( a r r 1 , a r r 2 ) (arr_1, arr_2) (arr1,arr2) 满足以下条件，我们称它们是单调数组对： 两个数组的长度都是 n n n。 a r r 1 arr_1 arr1 是…...

编程日记 2025/5/14 21:25:05

C++语法·叭

阁下何不乘风起，扶摇直上九万里。 qi fei 目录内存管理分区介绍 1.栈区： 2.内存映射段： 3.堆： 4.数据段： 5.代码段： 补充： C内存管理（简略回忆） C内存…...

编程日记 2025/5/18 8:35:34

ComfyUI | ComfyUI桌面版发布，支持winmac多平台体验，汉化共享等技巧！（内附安装包）

ComfyUI 桌面版正式推出，支持 Windows 与 macOS 等多平台，为 AI 绘画爱好者带来全新体验。其安装包便捷易用，开启了轻松上手之旅。汉化共享功能更是一大亮点，打破语言障碍，促进知识交流与传播。在操作上，它…...

编程日记 2025/5/15 1:00:57

探索Python词云库WordCloud的奥秘

文章目录探索Python词云库WordCloud的奥秘1. 背景介绍：为何选择WordCloud？2. WordCloud库简介3. 安装WordCloud库4. 简单函数使用方法5. 应用场景示例6. 常见Bug及解决方案7. 总结探索Python词云库WordCloud的奥秘 1. 背景介绍：为何选择Wo…...

编程日记 2025/5/17 23:42:24

用PHP抓取HTTPS资源时的常见问题与解决方法

概述随着互联网的发展，HTTPS已经成为主流协议，网站的数据安全性得到了显著提升。然而，对于开发者来说，HTTPS的广泛应用也增加了数据抓取的复杂性。尤其是在PHP中实现HTTPS资源的抓取时，开发者可能会遇到以下问题&…...

编程日记 2025/5/10 21:55:44

spring知识点复习--针对面试的

前言此内容是笔者通过B站的视频总结而来。原视频链接地址：6、Bean Factory与FactoryBean有什么区别_哔哩哔哩_bilibili 1.谈谈springIOC的理解，原理与实现回答涉及到的点： 控制反转：是一种理论思想，原来的对象是由…...

编程日记 2025/5/17 22:56:43

Web前端学习_CSS盒子模型

content padding border margin <!DOCTYPE html> <html lang"en"> <head><meta charset"UTF-8"><title>CSS盒子模型</title><style></style> </head> <body> <div class"demo&quo…...

编程日记 2025/5/14 5:56:50

一：express框架在终端输入nodejs文件名 // 引入express const express require(express); //创建应用对象 const app express(); //创建路由规则 app.get(/,(request,response) > {//设置响应response.send(Hello Express); }); // 监听3000端口 app.lis…...

编程日记 2025/5/15 7:04:02

Python轴承故障诊断 (21)基于VMD-CNN-BiTCN的创新诊断模型

往期精彩内容： Python-凯斯西储大学（CWRU）轴承数据解读与分类处理 Pytorch-LSTM轴承故障一维信号分类(一)-CSDN博客 Pytorch-CNN轴承故障一维信号分类(二)-CSDN博客 Pytorch-Transformer轴承故障一维信号分类(三)-CSDN博客三十多个开源…...

编程日记 2025/5/17 22:57:10

强化学习导论 -章9 基于函数逼近的同轨策略预测

基于函数逼近的同轨策略预测我们前面已经完成了基于表格的学习任务，基于表格的就是每个s是独立学习的，基本上不考虑泛化的能力，但是也对于每个任务状态学习的非常好。考虑到状态空间越来越大，我们必须考虑到函数逼近的情况。 1…...

编程日记 2025/5/16 19:39:46

Ubuntu环境中RocketMQ安装教程

参考教程 https://blog.csdn.net/weixin_56219549/article/details/126143231 1、安装JDK，并配置环境变量（略） 2、下载RocketMQ安装包 RocketMQ下载地址，选择二进制包下载 unzip rocketmq-all-5.0.0-ALPHA-bin-release.zip 使…...

编程日记 2025/5/14 10:05:08

Linux操作系统2-进程控制3(进程替换，exec相关函数和系统调用)

上篇文章：Linux操作系统2-进程控制2(进程等待，waitpid系统调用，阻塞与非阻塞等待)-CSDN博客本篇代码Gitee仓库：Linux操作系统-进程的程序替换学习 d0f7bb4 橘子真甜/linux学习 - Gitee.com 本篇重点：进程替换目录 …...

编程日记 2025/5/10 2:03:13

ThinkPHP Nginx 重写配置

目录 NGINX 重写 Admin项目隐藏入口文件，且禁用Admin模块&Admin.php 1️⃣配置仅用模块 2️⃣新增admin_xyz.php文件（自定义入口文件名），并绑定admin模块 3️⃣配置nginx 重写规则 NGINX 重写在Nginx低版本中&#xff0…...

编程日记 2025/5/13 4:51:58

SpringBoot小知识(2)：日志

日志是开发项目中非常重要的一个环节，它是程序员在检查程序运行的手段之一。 1.日志的基础操作 1.1 日志的作用编程期调试代码运营期记录信息： * 记录日常运营重要信息(峰值流量、平均响应时长……) * 记录应用报错信息(错误堆栈) * 记录运维过程数据(…...

编程日记 2025/5/13 7:46:18

深度学习：利用GPU进行训练

深度学习：利用GPU进行训练在现代深度学习框架中，如PyTorch，利用GPU加速模型训练是一种常见的做法。GPU（图形处理单元）由于其并行处理能力，特别适合执行大量的矩阵运算，这在训练神经网络时尤为…...

编程日记 2025/5/12 13:06:00

PHP 生成分享海报

因为用户端有多个平台，如果做分享海报生成，需要三端都来做，工作量比较大。所以这个艰巨的任务就光荣的交给后端了。经过一定时间的研究和调试，最终圆满完成了任务，生成分享海报图片实现笔记如下。目录准备字体文件…...

编程日记 2025/5/18 2:47:09

A050-基于spring boot物流管理系统设计与实现

🙊作者简介：在校研究生，拥有计算机专业的研究生开发团队，分享技术代码帮助学生学习，独立完成自己的网站项目。代码可以查看文章末尾⬇️联系方式获取，记得注明来意哦~🌹 赠送计算机毕业设计600…...

编程日记 2025/5/17 21:18:41

【docker】8. 镜像仓库实战

综合实战一：搭建一个 nginx 服务 Web 服务器 Web 服务器，一般是指“网站服务器”，是指驻留于互联网上某种类型计算机的程序。Web 服务器可以向 Web 浏览器等客户端提供文档，也可以放置网站文件，让全世界浏览&#xf…...

编程日记 2025/5/16 8:49:31

基于Springboot在线招投标系统【附源码】

基于Springboot在线招投标系统效果如下： 系统主页面系统登陆页面招标人信息管理页面招标信息管理页面招标信息页面评标信息页面中标信息页面研究背景随着互联网技术的不断发展，传统的招投标方式已经无法满足现代企业的需求。传统的招投标流程…...

编程日记 2025/5/16 2:50:15

elasticsearch集群部署及加密通讯

原文地址：elasticsearch集群部署及加密通讯 – 无敌牛欢迎参观我的个人博客：无敌牛 – 技术/著作/典籍/分享等第零步，准备给各台设备配置虚拟主机名，这样集群不依赖IP，即使IP变动，改动也更方便。参考…...

编程日记 2025/5/16 16:00:42

yolov5的pt模型转化为rk3588的rknn，并在rk3588上调用api进行前向推理

当使用yolov5进行目标检测且进行边缘计算的场景时，要考虑性价比或者国产化的话，rk3588板子是个不错的选择。本篇介绍yolov5的pytorch模型转化为rknn的流程，并展示在rk板子上如何调用相关api来使用转好的rknn模型进行前向推理。 pt转rknn流程…...

编程日记 2025/5/13 6:07:40

【机器学习】—PCA（主成分分析）

主成分分析（PCA）详解引言主成分分析（PCA）是一种统计方法，它可以通过正交变换将一组可能相关的变量转换为一组线性不相关的变量，这些变量称为主成分。PCA经常用于降维，数据压缩，以…...

编程日记 2025/5/15 10:42:23

【Linux】vim

🌻个人主页：路飞雪吖~ 🌠专栏：Linux 目录一、Linux开发工具 🌟vim的基本概念二、Linux编译器-gcc/g使用 🌟gcc如何完成（ESc - iso） 1、预处理（进行宏替换&#xff…...

编程日记 2025/5/18 6:55:03

virtualbox给Ubuntu22创建共享文件夹

1.在windows上的操作，创建共享文件夹Share 2.Ubuntu22上的操作，创建共享文件夹LinuxShare 3.在virtualbox虚拟机设置里，设置共享文件夹共享文件夹路径：选择Windows系统中你需要共享的文件夹共享文件夹名称：挂载至wi…...

编程日记 2025/5/12 14:40:39

SQLModel与FastAPI结合：构建用户增删改查接口

SQLModel简介 SQLModel是一个现代化的Python库，旨在简化与数据库的交互。它结合了Pydantic和SQLAlchemy的优势，使得定义数据模型、进行数据验证和与数据库交互变得更加直观和高效。SQLModel由FastAPI的创始人Sebastin Ramrez开发，专为与FastA…...

编程日记 2025/5/18 7:44:41

数据库日志

MySQL中有哪些日志 1，redo log重做日志 redo log是物理机日志，因为它记录的是对数据页的物理修改，而不是SQL语句。作用是确保事务的持久性，redo log日志记录事务执行后的状态，用来恢复未写入 data file的已提交事务…...

编程日记 2025/5/17 13:29:51

力扣第 71 题简化路径

一、题目描述给定一个字符串 path，表示一个由目录名和斜杠 "/" 组成的绝对路径，请简化该路径，使其变为规范路径。在 Unix 风格的文件系统中： 一个点 "." 表示当前目录本身；两个点 "..&q…...

编程日记 2025/5/12 17:53:41

Android 性能优化：内存优化（理论篇）

内存作为App程序运行最重要的资源之一，需要运行过程中做到合理的资源分配与回收，不合理的内存占用轻则使得用户应用程序运行卡顿、ANR、黑屏，重则导致用户应用程序发生 OOM（out of memory）崩溃。喜马直播随着近些年的业…...

编程日记 2025/5/11 18:18:12

Flink四大基石之窗口（Window）使用详解

目录一、引言二、为什么需要 Window 三、Window 的控制属性窗口的长度（大小） 窗口的间隔四、Flink 窗口应用代码结构是否分组 Keyed Window --键控窗 Non-Keyed Window 核心操作流程五、Window 的生命周期分配阶段触发计算六、Wi…...

编程日记 2025/5/18 1:44:39

Easy Excel 通过【自定义批注拦截器】实现导出的【批注】功能

目录 Easy Excel 通过【自定义批注拦截器】实现导出的【批注】功能需求原型：相关数据：要导出的对象字段postman 格式导出对象VO 自定义批注拦截器业务代码： 拦截器代码解释：详细解释：格式优化： Easy Excel…...

编程日记 2025/5/14 10:28:10

PHP 去掉特殊不可见字符 “\u200e“

描述最近在排查网站业务时，发现有数据匹配失败的情况肉眼上完全看不出问题所在当把字符串【M24308/23-14F‎】复制出来发现末尾有个不可见的字符使用删除键或左右移动时才会发现最后测试通过 var_dump 打印发现这个"空字符"占了三个长度 &#xf…...

编程日记 2025/5/14 11:18:06

Flume和kafka的整合：使用Flume将日志数据抽取到Kafka中

文章目录 1、Kafka作为Source【数据进入到kafka中，抽取出来】2、kafka作为Sink 【数据从别的地方抽取到kafka里面】 1、Kafka作为Source【数据进入到kafka中，抽取出来】 kafka源 --> memory --> 控制台： a1.sources r1 a1.sinks k1…...

编程日记 2025/5/15 17:31:05

Flutter：启动屏逻辑处理02：启动页

启动屏启动之后，制作一个启动页面新建splash：view 视图中只有一张图片sliding.png就是我们的启动图 import package:flutter/material.dart; import package:get/get.dart; import index.dart; class SplashPage extends GetView<SplashController…...

编程日记 2025/5/14 10:22:50

【MySQL】自动刷新flush privileges命令

在 MySQL 中，执行 FLUSH PRIVILEGES 命令的主要作用是使权限表中的更改立即生效。下面是关于这个命令的一些关键点： 1. 什么是 FLUSH PRIVILEGES 当你使用 SET PASSWORD 或其他 SQL 语句直接修改了用户的密码或权限（例如，使用 U…...

编程日记 2025/5/16 17:17:07

2024免费天气接口（无废话版）

免费接口1：http://t.weather.sojson.com/api/weather/city/101030100 免费接口2：http://t.weather.itboy.net/api/weather/city/101030100 至于后面那个城市编码请自行查询：如图注意！！！ 点击下载时可能…...

编程日记 2025/5/9 13:25:35

fpga 时序分析基础

目录触发器的动态参数同步时序电路分析 1. 时钟脉冲的特性 2. 同步时序电路分析 Timing Analyzer的应用异步时序与亚稳态问题时序分析就是对时序电路进行时序检查，通过分析电路中所有寄存器之间的路径延迟以检查电路的传输延迟是否会导致触发器的建立时间…...

编程日记 2025/5/14 21:12:26

Laravel8.5+微信小程序实现京东商城秒杀方案

一、商品秒杀涉及的知识点鉴权策略封装掊口访问频次限制小程序设计页面防抖接口调用订单创建事务使用超卖防御二、订单库存系统方案（3种） 下单减库存优点是库存和订单的强一致性，商品不会卖超，但是可能导致恶意下单&#xff…...

编程日记 2025/5/14 22:09:43

Git——本地仓库链接并推送到多个远程仓库

步骤 1. 新建仓库init 或删除已有仓库远程链接 // 1.新建init git init// 2.已有仓库，查看链接的远程仓库 git remote -v// 3.已有远程连接仓库，需要删除连接 git remote rm origin(或对应远程仓库名) 2.新建远程仓库在gitee、github等托管平台创建…...

编程日记 2025/5/6 17:47:42

llama-factory 系列教程 (七)，Qwen2.5-7B-Instruct 模型微调与vllm部署详细流程实战

文章目录介绍llama-factory 安装装包下载模型微调模型数据集训练模型微调后的模型推理介绍时隔已久的 llama-factory 系列教程更新了。本篇文章是第七篇，之前的六篇，大家酌情选看即可。因为llama-factory进行了更新，我前面几篇文章的实…...

编程日记 2025/5/15 1:31:14

Spring Boot的理解

一、什么是Spring Boot? Spring Boot是一个用于构建基于Spring框架的应用程序的开源框架。它简化了Spring应用程序的开发过程，使开发者能够更容易地创建独立运行的、生产级别的Spring应用程序。Spring Boot提供了许多功能和约定，可以帮助开发者快速搭建…...

编程日记 2025/5/14 21:44:09

QT QFormLayout控件全面详解

本系列文章全面的介绍了QT中的57种控件的使用方法以及示例，包括 Button(PushButton、toolButton、radioButton、checkBox、commandLinkButton、buttonBox)、Layouts(verticalLayout、horizontalLayout、gridLayout、formLayout)、Spacers(verticalSpacer、horizontalSpacer)、…...

编程日记 2025/5/15 13:23:38

系统性能定时监控PythonLinux

系统性能定时监控 1.系统监控概述⽤Python来编写脚本简化⽇常的运维⼯作是Python的⼀个重要⽤途。在Linux下，有许多系统命令可以让我们时刻监控系统运⾏的状态，如 ps ， top ， free 等等。要获取这些系统信息，Python…...

编程日记 2025/5/17 4:17:39

python学习——列表的相关操作

在 Python 中，列表（list）是一种非常灵活的数据结构，可以用来存储一系列的元素。以下是列表的一些常见操作： 文章目录创建列表访问元素修改元素列表切片添加元素删除元素列表推导式其他操作pop基本用法指定索引使用场…...

编程日记 2025/5/16 4:46:43

HTML CSS 魔法秀：打造翻转卡片登录注册页面

这段 HTML 和 CSS 代码创建了一个具有翻转卡片效果的登录和注册页面。下面是对重点标签和 CSS 样式的解释和总结： 一键复制 <!DOCTYPE html> <html lang"en"><head><meta charset"UTF-8"><meta http-equiv"…...

编程日记 2025/5/17 1:13:53

Web day04 SpringBoot

目录 1.Spring概念： 2. spring程序快速入门： 3.HTTP协议： 特点： 基于TCP 协议： 基于请求响应模型： HTTP协议是无状态协议： 请求协议：为浏览器向服务器发出的消息获取请求数据…...

编程日记 2025/5/15 4:55:53

selinux和防火墙实验

1 、 selinux 的说明 SELinux 是 Security-Enhanced Linux 的缩写，意思是安全强化的 linux 。 SELinux 主要由美国国家安全局（ NSA ）开发，当初开发的目的是为了避免资源的误用。系统资源都是通过程序进行访问的，如…...

编程日记 2025/5/17 0:25:59

ClamAV 在 CentOS 的开发版本 `clamav-devel`

是的，ClamAV 在 CentOS 上有开发版本（通常称为 clamav-devel），它包含了开发 ClamAV 应用程序所需的头文件和库文件。以下是如何在 CentOS 上安装 ClamAV 及其开发版本的步骤。 ### 1. **安装 EPEL 仓库** ClamAV 通常在 EPEL&am…...

编程日记 2025/5/17 2:45:20

C++《二叉搜索树》

在初阶数据结构中我学习了树基础的概念以及了解了顺序结构的二叉树——堆和链式结构二叉树该如何实现，那么接下来我们将进一步的学习二叉树，在此会先后学习到二叉搜索树、AVL树、红黑树；通过这些的学习将让我们更易于理解后面set、map、哈希等…...

编程日记 2025/5/12 23:59:38

深度学习:Pytorch中的模型保存与加载方法

1. 保存和加载整个模型

方法描述

参数和设置

示例代码

优点

缺点

2. 仅保存和加载模型的状态字典

方法描述

参数和设置

示例代码

load_state_dict 方法详解

优点

缺点

比较异同

相关文章：

`load_state_dict` 方法详解