如何完整读取.pt文件并查看全部权重及指定层权重

1. 显示全部权重值（避免省略号截断） #

PyTorch默认会对大张量输出进行截断，只需修改全局打印配置即可完整显示：

import torch
# 设置打印选项，让张量完整输出
torch.set_printoptions(threshold=torch.inf)
# 读取权重文件
weights = torch.load('file_name.pt')
# 此时打印weights将显示全部内容
print(weights)

如果仅需临时查看单个张量的完整内容，也可以直接对目标张量执行打印，上述全局配置会生效。

2. 打印指定层的权重值 #

torch.load读取的权重文件通常是字典结构（对应模型的state_dict），可直接通过键名访问指定层的权重：

# 先查看所有权重的键名，确认目标层的准确名称
print(weights.keys())
# 打印指定层的权重，例如L2.0.weight
print(weights['L2.0.weight'])

若提示键不存在，需检查键名是否与模型定义的层级、名称完全匹配（注意大小写、嵌套结构）。

内容的提问来源于stack exchange，提问作者Shweta Kiran