From b7b5ab8974ecf3b49085f8f2922bdb9f188eb0ac Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Fri, 13 Dec 2024 06:56:16 +0000
Subject: [PATCH] fix ZeroDivisionError

---
 ppcls/engine/train/utils.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/ppcls/engine/train/utils.py b/ppcls/engine/train/utils.py
index 1d5c70b2e..8de516ff4 100644
--- a/ppcls/engine/train/utils.py
+++ b/ppcls/engine/train/utils.py
@@ -52,8 +52,9 @@ def log_info(trainer, batch_size, epoch_id, iter_id):
         for key in trainer.time_info
     ])
 
-    ips_msg = "ips: {:.5f} samples/s".format(
-        batch_size / trainer.time_info["batch_cost"].avg)
+    avg = trainer.time_info["batch_cost"].avg
+    ips = batch_size / avg if avg != 0 else float("inf")
+    ips_msg = "ips: {:.5f} samples/s".format(ips)
 
     global_epochs = trainer.config["Global"]["epochs"]
     eta_sec = (