Style

e6aeb0d3 · Lysandre · 95a26fcf · e6aeb0d3
Commit e6aeb0d3 authored May 22, 2020 by Lysandre
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

src/transformers/trainer.py src/transformers/trainer.py +1 -3

No files found.
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -494,9 +494,7 @@ class Trainer:
                    ):
                        logs: Dict[str, float] = {}
                        logs["loss"] = (tr_loss - logging_loss) / self.args.logging_steps
-                        logs["learning_rate"] = (
-                            scheduler.get_last_lr()[0]
-                        )
+                        logs["learning_rate"] = scheduler.get_last_lr()[0]
                        logging_loss = tr_loss

                        self._log(logs)