Linear Regression 1D: Training One Parameter

Objective

• How to create cost or criterion function using MSE (Mean Square Error).

Table of Contents

In this lab, you will train a model with PyTorch by using data that you created. The model only has one parameter: the slope.

• Make Some Data
• Create the Model and Cost Function (Total Loss)
• Train the Model

Estimated Time Needed: 20 min