"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "25848a6094179391b03b34d447baa3aad4d815af"
Unverified Commit 02fe3af2 authored by Tom Aarsen's avatar Tom Aarsen Committed by GitHub
Browse files

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

Prevent ZeroDivisionError if evaluation is too quick
parent d924390d
...@@ -350,6 +350,8 @@ def speed_metrics(split, start_time, num_samples=None, num_steps=None): ...@@ -350,6 +350,8 @@ def speed_metrics(split, start_time, num_samples=None, num_steps=None):
""" """
runtime = time.time() - start_time runtime = time.time() - start_time
result = {f"{split}_runtime": round(runtime, 4)} result = {f"{split}_runtime": round(runtime, 4)}
if runtime == 0:
return result
if num_samples is not None: if num_samples is not None:
samples_per_second = num_samples / runtime samples_per_second = num_samples / runtime
result[f"{split}_samples_per_second"] = round(samples_per_second, 3) result[f"{split}_samples_per_second"] = round(samples_per_second, 3)
......
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