通过pytorch使用python代码打开并读取PT.文件

qhhrdooz 于 2023-02-19 发布在 Python

关注(0)|答案(2)|浏览(1556)

我想用python读取一个PT文件，但我不知道怎么做，我想用python打开它
你能帮帮我吗，有什么主意吗？

2条答案

如果您的.PT文件与model的权重和偏差有关。
你必须先在你的电脑上安装pytorch。
有关详细信息，请转到this
然后使用以下命令：

model = torch.load(PATH)

saving_loading_models
您可以通过以下方式迭代参数以获得所有权重和偏差参数：（请参见权重和偏差）

for param in model.parameters():
   ....
# or
for name, param in model.named_parameters():
   ...

import torch

# Load the file
pt_file = torch.load("./myfile.pt")

# Print the head of the file
print(pt_file[:5])