From c4020147a3cf28d6c47c6770f8de6f572191ca16 Mon Sep 17 00:00:00 2001 From: Isak Falk Date: Fri, 23 Oct 2020 15:15:28 +0100 Subject: [PATCH] Fix double error normalisation in maml example (#196) --- examples/vision/maml_miniimagenet.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/examples/vision/maml_miniimagenet.py b/examples/vision/maml_miniimagenet.py index c57fd8d1..d2e06681 100644 --- a/examples/vision/maml_miniimagenet.py +++ b/examples/vision/maml_miniimagenet.py @@ -43,13 +43,11 @@ def fast_adapt(batch, learner, loss, adaptation_steps, shots, ways, device): # Adapt the model for step in range(adaptation_steps): adaptation_error = loss(learner(adaptation_data), adaptation_labels) - adaptation_error /= len(adaptation_data) learner.adapt(adaptation_error) # Evaluate the adapted model predictions = learner(evaluation_data) evaluation_error = loss(predictions, evaluation_labels) - evaluation_error /= len(evaluation_data) evaluation_accuracy = accuracy(predictions, evaluation_labels) return evaluation_error, evaluation_accuracy