Tutorial 3: how to train a liear regression model using nn.Linear()