Exercise

Implementing the training step

In this exercise, you'll implement the training_step() method in a PyTorch Lightning module designed for an image classification task. Your implementation should unpack a batch of images and labels, compute the model predictions via the forward pass, calculate the cross entropy loss, and log the training loss.

Instructions

100 XP

Ensure that you compute predictions using the forward pass.
Calculate the cross entropy loss.
Log the training loss.

.css-6su6fj{-webkit-flex-shrink:0;-ms-flex-negative:0;flex-shrink:0;}Exercise

Instructions

Exercise