diff --git a/optimus/trainer.py b/optimus/trainer.py index 065d61a49f2a9c3c7ef57f1d51c935f0e2e0e97b..f55f73ebb1532ca67869bc0244a7723b371b9f8f 100644 --- a/optimus/trainer.py +++ b/optimus/trainer.py @@ -155,10 +155,6 @@ class Trainer(): f"~{self.ms_per_batch:.2f} ms/batch | " \ f" lr: {lr:.7f}" - # account for last batches when computing average train loss - self.train_loss = total_loss / (len(self.dl.train) % est_interval - 1) - self.train_ppl = math.exp(self.train_loss) - pb.on_iter_end() def _do_epoch_validate(self):