riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 18:22:34 +06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Workaround for #27758 to avoid ZeroDivisionError (#28756)

Browse Source
This commit is contained in:
Traun Leyden 2024-03-04 10:23:40 +01:00 committed by GitHub
parent 704b3f74f9
commit c38a12270a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/trainer.py
View File

@ -2080,7 +2080,8 @@ class Trainer:
# add remaining tr_loss # add remaining tr_loss
self._total_loss_scalar += tr_loss.item() self._total_loss_scalar += tr_loss.item()
train_loss = self._total_loss_scalar / self.state.global_step effective_global_step = max(self.state.global_step, 0.001) # Avoid ZeroDivisionError
train_loss = self._total_loss_scalar / effective_global_step
metrics = speed_metrics( metrics = speed_metrics(
"train", "train",