小编Muh*_*mel的帖子

我正在构建一个神经网络，但我不知道如何访问每一层的模型权重。

我试过了

model.input_size.weight

代码：

input_size = 784
hidden_sizes = [128, 64]
output_size = 10

# Build a feed-forward network
model = nn.Sequential(nn.Linear(input_size, hidden_sizes[0]),
                      nn.ReLU(),
                      nn.Linear(hidden_sizes[0], hidden_sizes[1]),
                      nn.ReLU(),
                      nn.Linear(hidden_sizes[1], output_size),
                      nn.Softmax(dim=1))

我希望得到重量，但我得到了

“顺序”对象没有属性“input_size”