how to

5.2.参数管理

Aug 23, 2024
notesjulyfun技术学习d2l
1 Minutes
52 Words

https://zh.d2l.ai/chapter_deep-learning-computation/parameters.html

1
net = nn.Sequential(nn.Linear(4, 8), nn.ReLU(), nn.Linear(8, 1))
2
X = torch.rand(size=(2, 4))
3
net(X)

访问参数

  • net[2].state_dict(): 第三层参数字典
  • net[2].bias net[2].bais.data: 第三层偏置捏
  • net.named_parameters(): 所有参数字典
  • net.state_dict()['2.bias'].data: 第三层偏置
  • print(net): 显示网络结构
Article title:5.2.参数管理
Article author:Julyfun
Release time:Aug 23, 2024
Copyright 2025
Sitemap