Created 4 years ago
![cognitiveclass.ai logo](https://s3-api.us-geo.objectstorage.softlayer.net/cf-courses-data/CognitiveClass/DL0110EN/notebook_images%20/cc-logo-square.png)
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