Fix typo when loading train state (#2425)

* Fix typo when loading train state * Fix comment
allenai · Jan 23, 2019 · b7d56ae · b7d56ae
1 parent cb9651a
commit b7d56ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/allennlp/training/trainer.py b/allennlp/training/trainer.py
@@ -565,10 +565,10 @@ def _restore_checkpoint(self) -> int:
         # Currently the ``training_state`` contains a serialized ``MetricTracker``.
         if "metric_tracker" in training_state:
             self._metric_tracker.load_state_dict(training_state["metric_tracker"])
-        # It used to be the case that we tracked ``val_metrics_per_epoch``.
+        # It used to be the case that we tracked ``val_metric_per_epoch``.
         elif "val_metric_per_epoch" in training_state:
             self._metric_tracker.clear()
-            self._metric_tracker.add_metrics(training_state["val_metrics_per_epoch"])
+            self._metric_tracker.add_metrics(training_state["val_metric_per_epoch"])
         # And before that we didn't track anything.
         else:
             self._metric_tracker.clear()