MulaiMulai sekarang secara gratis

Menulis evaluation loop

Dalam latihan ini, Anda akan menulis evaluation loop untuk menghitung validation loss. Evaluation loop memiliki struktur yang mirip dengan training loop namun tanpa perhitungan gradien atau pembaruan bobot.

model, validationloader, dan fungsi loss criterion sudah didefinisikan untuk menangani prediksi, pemuatan data, dan perhitungan loss.

Latihan ini adalah bagian dari kursus

Pengantar Deep Learning dengan PyTorch

Lihat Kursus

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# 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 += ____
Edit dan Jalankan Kode