Open Iwillsd opened 2 years ago
x = torch.randn(10, 3) y = torch.randn(10, 2)
linear = nn.linear(3, 2) print('w:', linear.weight) print('b:', linear.bias)
loss = nn.MSELoss() optimizer = torch.optim.SGD(linear.parameters(), lr=0.01)
Please help answer this question, thank you very much!
It's nn.Linear, with a capital L.
linear = nn.Linear(3, 2)
it's with capital L in Linear isn't it? If it's the case, I think this issue should be closed.
=============================================
2.Basic autogard example2
=============================================
Create tensors of shape (10,3) and (10,2)
x = torch.randn(10, 3) y = torch.randn(10, 2)
Build a fully connected layer
linear = nn.linear(3, 2) print('w:', linear.weight) print('b:', linear.bias)
Build loss function and optimizer
loss = nn.MSELoss() optimizer = torch.optim.SGD(linear.parameters(), lr=0.01)
Please help answer this question, thank you very much!