Linear Regression 1D: Training Two Parameter Mini-Batch Gradient Decent

Objective

• How to use Mini-Batch Gradient Descent to train model.

Table of Contents

In this Lab, you will practice training a model by using Mini-Batch Gradient Descent.

• Make Some Data
• Create the Model and Cost Function (Total Loss)
• Train the Model: Batch Gradient Descent
• Train the Model: Stochastic Gradient Descent with Dataset DataLoader
• Train the Model: Mini Batch Gradient Decent: Batch Size Equals 5
• Train the Model: Mini Batch Gradient Decent: Batch Size Equals 10

Estimated Time Needed: 30 min