Fixed divide by zero when tuning batch size (#3481)

ludwig-ai · Jul 26, 2023 · 446c747 · 446c747
1 parent 279a9ba
commit 446c747
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ludwig/utils/batch_size_tuner.py b/ludwig/utils/batch_size_tuner.py
@@ -100,7 +100,11 @@ def evaluate(self, batch_size: int, total_steps: int = 5) -> float:
  start_ts = time.time()
  self.step(batch_size)
  durations.append(time.time() - start_ts)
+
  med_duration_s = statistics.median(durations)
+ if med_duration_s == 0.0:
+ return float("inf")
+
  return batch_size / med_duration_s
 
  def reset(self):