Discussion 4 Pytorch
Discussion 4 Pytorch
Xiao et al. Fashion-mnist: a novel image dataset for benchmarking machine learning algorithms.
arXiv preprint arXiv:1708.07747.
Tensors
Initializing a Tensor
Initializing a Tensor
Attributes of a Tensor
Operations on Tensors
Operations on Tensors
Operations on Tensors
Datasets & DataLoaders
Loading a Dataset
Iterating and Visualizing the Dataset
Iterating and Visualizing the Dataset
Creating a Custom Dataset for your files
Creating a Custom Dataset for your files
Creating a Custom Dataset for your files
Creating a Custom Dataset for your files
Preparing your data for training with
DataLoaders
Iterate through the DataLoader
Iterate through the DataLoader
Neural Network
torch.nn.Module
Get Device for Training
Define the Class
Define the Class
Define the Class
Optimizing Model Params
Loss Function
Optimizer
Full Implementation – Train Loop
Full Implementation – Test Loop
Full Implementation
Full Implementation
Save and Load the Model
Saving and Loading Model Weights
Saving and Loading Models with Shapes
Acknowledgment