Image Classification and CNN
Building a CNN Classifier with PyTorch: Part 1
Building a CNN Classifier with PyTorch: Part 2
Introduction to DDP with PyTorch
Single Node Multi-GPU Training with torchrun
Multi-node Distributed Training with PyTorch
Lab Setup

Dataset Loaders and Transforms

In Part1 of building a CNN Classifier, we introduced three PyTorch tools:

  • Dataset: stores data and their corresponding labels
  • Transforms: performs data manipulation to make data suitable for training
  • DataLoaders: iterable around the dataset for ease of access to samples from the dataset.

In this notebook, we will expand on additional transforms that can be done to improve the performances of a CNN image classifier.

Back
 
Next
© Chishiki-AI  |   Cornell University    |   Center for Advanced Computing    |   Copyright Statement    |   Access Statement
CVW material development is supported by NSF OAC awards 1854828, 2321040, 2323116 (UT Austin) and 2005506 (Indiana University)