yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AttributeError: module 'torch.nn' has no attribute 'linear'

Iwillsd opened this issue · comments

#=============================================#

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!

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.