Skriva evalueringsloopen
I den här övningen skriver du en evalueringsloop för att beräkna valideringsförlusten. Evalueringsloopen följer en liknande struktur som träningsloopen, men utan gradientberäkningar eller viktuppdateringar.
model, validationloader och förlustfunktionen criterion har redan definierats för att hantera prediktioner, dataladdning och förlustberäkning.
Den här övningen är en del av kursen
Introduktion till djupinlärning med PyTorch
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# 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 += ____