【PyTorch】.pt文件

ops/2025/3/19 6:22:53/

.pt文件是 PyTorch 中用于保存张量(torch.Tensor)或模型(torch.nn.Module)的二进制文件格式。它使用 PyTorch 的序列化机制来保存数据,能够高效地存储和加载张量或模型的状态。

.pt 文件中存储的内容

1. 张量(torch.Tensor)
如果保存的是张量(如 y ),.pt文件会存储张量的以下信息:

  • 张量的数据(数值)。
  • 张量的形状(shape)。
  • 张量的数据类型(dtype,如 float32、int64等)。
  • 张量的设备信息(device,如 cpu或 cuda)。

2. 模型(torch.nn.Module)

  • 如果保存的是模型,.pt文件会存储模型的以下信息:
  • 模型的参数(state_dict)。
  • 模型的结构(如果使用 torch.save(model, ...))。
  • 优化器的状态(如果同时保存优化器)。

3. 其他 Python 对象
.pt文件还可以保存其他 Python 对象(如字典、列表等),只要这些对象可以被 PyTorch 的序列化机制处理。

如何查看 .pt文件的内容

要查看 .pt 文件的内容,可以使用 torch.load加载文件,然后打印或检查加载的对象。

python">import torch# 加载 .pt 文件
y = torch.load("y_batch_0.pt")
y_likelihoods = torch.load("y_likelihoods_batch_0.pt")# 查看张量的信息
print("y:", y)
print("y shape:", y.shape)
print("y dtype:", y.dtype)
print("y device:", y.device)

http://www.ppmy.cn/ops/166962.html

相关文章

pnpm config set ignore-workspace-root-check true

异常 ERR_PNPM_ADDING_TO_ROOT  Running this command will add the dependency to the workspace root, which might not be what you want - if you really meant it, make it explicit by running this command again with the -w flag (or --workspace-root). If you don…

【后端】【django】Django DRF `@action` 详解:自定义 ViewSet 方法

Django DRF action 详解:自定义 ViewSet 方法 在 Django REST Framework(DRF)中,action 装饰器用于为 ViewSet 添加自定义的 API 端点。相比于 update、create 等默认方法,action 允许我们定义 更加清晰、语义化 的 A…

springboot基于session实现登录

文章目录 1.理解session2.理解ThreadLocal2.1 理解多线程2.2 理解lambda表达式2.3 ThreadLocal 3.基于session登录流程图4.具体登录的代码实现4.1短信发送功能4.2 短信验证码登录注册功能4.登录校验功能4.1 配置登录拦截器LoginInterceptor4.1.1 ThrealLocal类实现 4.2登录拦截…

EditRocket for Mac v5.0.2 文本编辑器 支持M、Intel芯片

应用介绍 EditRocket 是一款强大的跨平台文本编辑器,专为程序员、开发者和技术人员设计,提供了丰富的编程支持和多种开发工具。它支持各种编程语言,具备高效的代码编辑、调试、和文本处理功能,旨在提升编程效率和开发体验。 主要…

c语言数据结构——单向不带头不循环链表的实现

文章目录 单向不带头不循环链表链表与顺序表的区别多文件管理链表的定义结构获得链表节点个数链表增加元素链表的尾插及创建节点函数链表的头插任意位置节点后插入 判断链表是否为空链表删除元素链表的尾删链表的头删任意位置删除 链表查找元素链表修改元素单向链表的遍历链表销…

reactive数据修改无效

环境 vue:3.2.13 element-plus: 2.9.6 typescript:4.5.5 问题 表格列表页面,页面中有新增和修改操作,新增和修改共用一个弹窗,弹窗中表单绑定的数据修改无效。复现步骤是先点击表格中的修改,然后点击新…

jasypt-spring-boot-starter项目如何使用jasypt加密密码

import org.jasypt.encryption.pbe.StandardPBEStringEncryptor; import org.jasypt.iv.RandomIvGenerator; import org.jasypt.salt.RandomSaltGenerator;/*** 加密密码的工具** author xxx* since 2025-03-17*/ public class JasyptTest {public static void main(String[] a…

汽车机械钥匙升级一键启动的优点

汽车机械钥匙升级一键启动的优点主要包括: 便捷性:一键启动功能的引入极大地提升了用车便捷性。车主无需翻找钥匙,只需在车辆感应范围内轻触启动键,即可轻松发动汽车。 安全性:移动管家专车专用一键启动系统配备了防…