Fix omniglot double normalization issue (#201)

learnables · Nov 11, 2020 · 9af5ee7 · 9af5ee7
1 parent c402014
commit 9af5ee7
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/examples/vision/maml_omniglot.py b/examples/vision/maml_omniglot.py
@@ -36,13 +36,11 @@ def fast_adapt(batch, learner, loss, adaptation_steps, shots, ways, device):
     # Adapt the model
     for step in range(adaptation_steps):
         train_error = loss(learner(adaptation_data), adaptation_labels)
-        train_error /= len(adaptation_data)
         learner.adapt(train_error)
 
     # Evaluate the adapted model
     predictions = learner(evaluation_data)
     valid_error = loss(predictions, evaluation_labels)
-    valid_error /= len(evaluation_data)
     valid_accuracy = accuracy(predictions, evaluation_labels)
     return valid_error, valid_accuracy