Skip to content

GitLab

Commit e6aeb0d3 authored by Lysandre's avatar Lysandre
Browse files

Style

parent 95a26fcf
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
src/transformers/trainer.py
View file @ e6aeb0d3
...@@ -494,9 +494,7 @@ class Trainer: ...@@ -494,9 +494,7 @@ class Trainer:
): ):
logs: Dict[str, float] = {} logs: Dict[str, float] = {}
logs["loss"] = (tr_loss - logging_loss) / self.args.logging_steps logs["loss"] = (tr_loss - logging_loss) / self.args.logging_steps
logs["learning_rate"] = ( logs["learning_rate"] = scheduler.get_last_lr()[0]
scheduler.get_last_lr()[0]
)
logging_loss = tr_loss logging_loss = tr_loss
self._log(logs) self._log(logs)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment