Fix double error normalisation in maml example (#196)

learnables · Oct 23, 2020 · c402014 · c402014
1 parent 76fb944
commit c402014
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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