Pisanie pętli ewaluacyjnej
W tym ćwiczeniu napiszesz pętlę ewaluacyjną do obliczania straty walidacyjnej. Pętla ewaluacyjna ma podobną strukturę do pętli treningowej, ale bez obliczania gradientów i aktualizacji wag.
model, validationloader oraz funkcja straty criterion są już zdefiniowane – obsługują odpowiednio predykcje, ładowanie danych i obliczanie straty.
To ćwiczenie jest częścią kursu
Wprowadzenie do uczenia głębokiego z PyTorch
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Set the model to evaluation mode
____
validation_loss = 0.0
with torch.no_grad():
for features, labels in validationloader:
outputs = model(features)
loss = criterion(outputs, labels)
# Sum the current loss to the validation_loss variable
validation_loss += ____