AttributeError: module 'torch.nn' has no attribute 'linear'
Author: IwillsdCreated Aug 22, 2021Updated Nov 23, 2022
#=============================================#
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!
Source: yunjey/pytorch-tutorial