From e6f211cade5c69587a5e485499e39b4ceb57e63f Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Mon, 11 Jan 2021 10:17:49 -0800 Subject: [PATCH] [trainer] round numbers in trainer state (#9491) * round numbers * style * round only on logging --- src/transformers/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index 76e29a1ad5..b4abb8d333 100755 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -964,7 +964,7 @@ class Trainer: # reset tr_loss to zero tr_loss -= tr_loss - logs["loss"] = tr_loss_scalar / (self.state.global_step - self._globalstep_last_logged) + logs["loss"] = round(tr_loss_scalar / (self.state.global_step - self._globalstep_last_logged), 4) # backward compatibility for pytorch schedulers logs["learning_rate"] = ( self.lr_scheduler.get_last_lr()[0] @@ -1168,7 +1168,7 @@ class Trainer: The values to log. """ if self.state.epoch is not None: - logs["epoch"] = self.state.epoch + logs["epoch"] = round(self.state.epoch, 2) self.control = self.callback_handler.on_log(self.args, self.state, self.control, logs) output = {**logs, **{"step": self.state.global_step}}