refactor

app.py

@@ -99,6 +99,7 @@ class ProgressReport:
     def set_n_overall(self, total_overall_steps):
         self.current_overall_step = 0
         self.update_progress(overall_progress=0, overall_text="")
+        self.total_overall_steps = total_overall_steps
 
     def reset_batch(self, step_name=""):
         self.current_batch = 0