Implementace trénovacího kroku
V tomto cvičení implementuješ metodu training_step() v modulu PyTorch Lightning navrženém pro klasifikaci obrázků.
Tvoje implementace by měla rozbalit dávku obrázků a štítků, vypočítat predikce modelu pomocí dopředného průchodu, spočítat ztrátu cross entropy a zalogovat trénovací ztrátu.
Toto cvičení je součástí kurzu
Škálovatelné modely AI s PyTorch Lightning
Pokyny k cvičení
- Predikce vypočítej pomocí dopředného průchodu.
- Vypočítej ztrátu cross entropy.
- Zaloguj trénovací ztrátu.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
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