Eğitim adımını uygulama
Bu egzersizde, bir görüntü sınıflandırma görevi için tasarlanmış bir PyTorch Lightning modülünde training_step() metodunu uygulayacaksın.
Uygulaman, bir görüntü ve etiket yığınını açmalı, forward geçişiyle model tahminlerini hesaplamalı, cross entropy kaybını hesaplamalı ve eğitim kaybını kayıt altına almalıdır.
Bu egzersiz
PyTorch Lightning ile Ölçeklenebilir AI Modelleri
kursunun bir parçasıdırEgzersiz talimatları
- Tahminleri forward geçişini kullanarak hesapladığından emin ol.
- Cross entropy kaybını hesapla.
- Eğitim kaybını kaydet.
Uygulamalı interaktif egzersiz
Bu örnek kodu tamamlayarak bu egzersizi bitirin.
from torch.nn.functional import cross_entropy
def training_step(self, batch, batch_idx):
x, y = batch
# Ensure that you compute predictions using the forward pass
y_hat = ____
# Calculate the cross entropy loss
loss = ____
# Log the loss
self.____("train_loss", loss)
return loss