Completed
Mental model: Training a neural network in PyTorch
Class Central Classrooms beta
YouTube videos curated by Class Central.
Classroom Contents
Train a PyTorch Neural Network Step-by-Step - PyTorch Deep Learning
Automatically move to the next video in the Classroom when playback concludes
- 1 Mental model: Training a neural network in PyTorch
- 2 PyTorch imports and hyperparemeters
- 3 Training and test data for our FashionMNIST classification
- 4 Creating our Neural Network using PyTorch's nn module
- 5 Loss function, Cross-Entropy Loss and Optimizers
- 6 Setting up a training loop for the forward pass + backward pass
- 7 Visualize our computation graph with pytorchviz
- 8 Saving and loading pytorch models