文章目录
简介
此篇博客简单讲解如何使用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:一般参数值非常多,此处仅截取了部分已做展示❗️】
收集整理和创作不易, 若有帮助🉑, 请帮忙点赞
👍➕收藏
❤️, 谢谢!✨✨🚀🚀