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, kursun bir parçasıdır
PyTorch Lightning ile Ölçeklenebilir AI Modelleri
Egzersiz talimatları
- Tahminleri forward geçişini kullanarak hesapladığından emin ol.
- Cross entropy kaybını hesapla.
- Eğitim kaybını kaydet.
Uygulamalı etkileşimli egzersiz
Bu egzersizi bu örnek kodu tamamlayarak deneyin.
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