Training image classification model using python (pytorch)

Learn how to train an image classifier using python + pytorch. We will use a pretrained model (Resnet18) to classify animal images. The same process can be used to train a classifier for different contexts like medicine (radiology), autonomous vehicles, etc.

00:00 Introduction and mindset
04:50 Set up 1 (VScode + python and jupyter extensions)
09:00 Set up 2 (Creating a virtual environment and installing libraries)
12:50 Preparing the dataset
15:50 Inspecting the data
17:28 Defining/Loading Neural Net Architecture
18:50 Loss function + Optimizer
19:57 Training Loop
22:55 Visualizing performance
24:00 Training vs Validation vs Testing
27:50 Saving the weights of the model
28:16 Loading the weights and predicting on sample images