fix logging error (#1017)

allenai · Mar 22, 2018 · 3ea3e64 · 3ea3e64
1 parent ab3bd0f
commit 3ea3e64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/allennlp/training/trainer.py b/allennlp/training/trainer.py
@@ -460,7 +460,7 @@ def _train_epoch(self, epoch: int) -> Dict[str, float]:
                 self._optimizer.step()
 
             # Update the description with the latest metrics
-            metrics = self._get_metrics(train_loss, self._batch_num_total)
+            metrics = self._get_metrics(train_loss, batches_this_epoch)
             description = self._description_from_metrics(metrics)
 
             train_generator_tqdm.set_description(description, refresh=False)