src/transformers/debug_utils.py/DebugUnderflowOverflow/batch_end_frame
class DebugUnderflowOverflow:    def batch_end_frame(self):
        self.expand_frame(f"{self.prefix} *** Finished batch number={self.batch_number-1} ***\n\n")