1. Nauka
  2. /
  3. Kursy
  4. /
  5. Skalowalne modele AI z PyTorch Lightning

Connected

ćwiczenie

Ocena dokładności modelu za pomocą Torchmetrics

Ocena skuteczności modelu to kluczowy krok – szczególnie gdy przygotowujesz go do wdrożenia! Zintegrujmy obliczanie dokładności za pomocą Torchmetrics bezpośrednio w metodzie validation_step(). Pamiętaj, aby rejestrować wyniki – dzięki temu łatwo będziesz mógł śledzić postępy modelu.

Instrukcje

100 XP
  • Zaimportuj Accuracy z torchmetrics.
  • Utwórz instancję metryki dokładności wewnątrz __init__().
  • Oblicz dokładność w metodzie validation_step() i zapisz ją jako 'val_acc'.