mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix ZeroDivisionError

pull/3324/head
gaotingquan 2024-12-13 06:56:16 +00:00
parent 0615d4beb3
commit b7b5ab8974
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ppcls/engine/train/utils.py
View File

@ -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 = (