Update program.py (#13989)

avoid ZeroDivisionError when training.
PaddlePaddle · Oct 13, 2024 · ea7ba39 · ea7ba39
1 parent 9668890
commit ea7ba39
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tools/program.py b/tools/program.py
@@ -709,7 +709,11 @@ def eval(
 
     pbar.close()
     model.train()
-    metric["fps"] = total_frame / total_time
+    # Avoid ZeroDivisionError
+    if total_time > 0:
+        metric["fps"] = total_frame / total_time
+    else:
+        metric["fps"] = 0  # or set to a fallback value
     return metric