简介

此篇博客简单讲解如何使用Python脚本查看PyTorch模型的参数。

Python查看参数的脚本

在PyTorch中，你可以使用state_dict()方法来查看模型的状态字典，它包含了模型的权重和偏置。🚀🚀🚀

import torch
import torch.nn as nn
 
# 定义一个简单的模型
class SimpleModel(nn.Module):
    def __init__(self):
        super(SimpleModel, self).__init__()
        self.conv1 = nn.Conv2d(1, 3, 5)
        self.fc1 = nn.Linear(3 * 3 * 3, 10)
        self.fc2 = nn.Linear(10, 10)
 
    def forward(self, x):
        x = self.conv1(x)
        x = x.view(-1, 3 * 3 * 3)
        x = torch.relu(self.fc1(x))
        x = self.fc2(x)
        return x
 
# 实例化模型
model = SimpleModel()
 
# 分别查看模型的state_dict的键、值
print(model.state_dict().keys())
print(model.state_dict().items())

• 部分结果如下【PS：一般参数值非常多，此处仅截取了部分已做展示❗️】
  

收集整理和创作不易, 若有帮助🉑, 请帮忙点赞👍➕收藏❤️, 谢谢!✨✨🚀🚀