riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Style

Browse Source
This commit is contained in:
Lysandre 2020-05-22 17:20:03 -04:00
parent 95a26fcf2d
commit e6aeb0d3e8
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/trainer.py
View File

@ -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)