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

Prevent ZeroDivisionError on trainer.evaluate if model and dataset are tiny (#24049)

Browse Source 
Prevent ZeroDivisionError if evaluation is too quick
This commit is contained in:
Tom Aarsen 2023-06-06 17:31:05 +02:00 committed by GitHub
parent d924390d5b
commit 02fe3af275
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/trainer_utils.py
View File

@ -350,6 +350,8 @@ def speed_metrics(split, start_time, num_samples=None, num_steps=None):
"""
runtime = time.time() - start_time
result = {f"{split}_runtime": round(runtime, 4)}
if runtime == 0:
return result
if num_samples is not None:
samples_per_second = num_samples / runtime
result[f"{split}_samples_per_second"] = round(samples_per_second, 3)