AttributeError: module 'torch.nn' has no attribute 'linear'
Iwillsd opened this issue · comments
Iwillsd commented
#=============================================#
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!
Nikolai Gaiduchenko commented
It's nn.Linear, with a capital L.
linear = nn.Linear(3, 2)
Nour Soltani commented
it's with capital L in Linear isn't it?
If it's the case, I think this issue should be closed.