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

Conclusion

In this topic, we introduced the basics of how to build a CNN classification model using transfer learning as a fixed feature extractor. There were three major steps:

  1. Loading and transforming our data

  2. Building the architecture of our network

  3. Training our model

In part 2, we will make several modifications to the workflow introduced in this notebook and see if we can improve the performance of our model.

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)