Linear regression 1D: Training Two Parameter

Objective

• How to train the model and visualize the loss results.

Table of Contents

In this lab, you will train a model with PyTorch by using the data that we created. The model will have the slope and bias. And we will review how to make a prediction in several different ways by using PyTorch.

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

Estimated Time Needed: 20 min