【PyTorch】.pt文件

.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加载文件，然后打印或检查加载的对象。

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)