update

deependujha · deependujha · commit 340e1d896912 · 2025-07-25T23:42:04.000+05:30
diff --git a/src/lightning/pytorch/callbacks/progress/rich_progress.py b/src/lightning/pytorch/callbacks/progress/rich_progress.py
@@ -448,8 +448,6 @@ def _add_task(self, total_batches: Union[int, float], description: str, visible:
         )
 
     def _initialize_progress_bar_id(self) -> None:
-        # Initialize the training progress bar here because
-        # `on_train_epoch_start` is not called when resuming from a mid-epoch restart
         total_batches = self.total_train_batches
         train_description = self._get_train_description(self.trainer.current_epoch)
         assert self.progress is not None

Original file line number	Diff line number	Diff line change
`@@ -448,8 +448,6 @@ def _add_task(self, total_batches: Union[int, float], description: str, visible:`
`448`	`448`	`)`
`449`	`449`
`450`	`450`	`def _initialize_progress_bar_id(self) -> None:`
`451`		`- # Initialize the training progress bar here because`
`452`		- # `on_train_epoch_start` is not called when resuming from a mid-epoch restart
`453`	`451`	`total_batches = self.total_train_batches`
`454`	`452`	`train_description = self._get_train_description(self.trainer.current_epoch)`
`455`	`453`	`assert self.progress is not None`