diff --git a/mz_train_tools_utils.py b/mz_train_tools_utils.py index da336c8..b52a7b6 100644 --- a/mz_train_tools_utils.py +++ b/mz_train_tools_utils.py @@ -758,6 +758,7 @@ class Utils: step, total_steps, pil_img_info) except Exception as e: print("progress_bar:", e) + raise e return pb(steps)