Lightning AI Studios: Never set up a local environment again →

Log in or create a free Lightning.ai account to track your progress and access additional course materials  

3.0 Overview – Model Training in PyTorch

Congratulations on completing units 1 and 2. You now have a solid understanding of the core principles of machine learning. We also covered the computational basics and learned about using tensors in PyTorch. We will expand on these concepts and introduce additional PyTorch functions throughout the rest of this course.

Unit 3 introduces the concept of single-layer neural networks and a new classification model: logistic regression. While this course is about deep learning, logistic regression is an excellent model for studying all the fundamental concepts we will use to implement deep neural networks in the upcoming units.

Log in or create a free Lightning.ai account to access:

  • Quizzes
  • Completion badges
  • Progress tracking
  • Additional downloadable content
  • Additional AI education resources
  • Notifications when new units are released
  • Free cloud computing credits
Watch Video 1

Unit 3

Videos