Register batch complete progress before batch_end hooks #14107

Job	Run time
pl-cpu (macOS-14, lightning, 3.10, 2.1)	19m 23s
pl-cpu (ubuntu-20.04, lightning, 3.10, 2.1)	9m 34s
pl-cpu (windows-2022, lightning, 3.10, 2.1)	17m 53s
pl-cpu (macOS-14, lightning, 3.11, 2.2)	19m 33s
pl-cpu (ubuntu-20.04, lightning, 3.11, 2.2)	14m 21s
pl-cpu (windows-2022, lightning, 3.11, 2.2)	17m 11s
pl-cpu (macOS-14, lightning, 3.11, 2.3)	21m 5s
pl-cpu (ubuntu-20.04, lightning, 3.11, 2.3)	14m 28s
pl-cpu (windows-2022, lightning, 3.11, 2.3)	16m 54s
pl-cpu (macOS-14, lightning, 3.12, 2.4)	20m 5s
pl-cpu (ubuntu-20.04, lightning, 3.12, 2.4)	20m 50s
pl-cpu (windows-2022, lightning, 3.12, 2.4)	16m 36s
pl-cpu (macOS-12, pytorch, 3.10, 2.1)	31m 18s
pl-cpu (ubuntu-22.04, pytorch, 3.10, 2.1)	13m 49s
pl-cpu (windows-2022, pytorch, 3.10, 2.1)	16m 16s
pl-cpu (macOS-13, lightning, 3.9, 2.1, oldest)	25m 57s
pl-cpu (ubuntu-20.04, lightning, 3.9, 2.1, oldest)	12m 10s
pl-cpu (windows-2022, lightning, 3.9, 2.1, oldest)	14m 47s
pl-cpu (macOS-14, pytorch, 3.9, 2.1)	22m 52s
pl-cpu (ubuntu-20.04, pytorch, 3.9, 2.1)	14m 18s
pl-cpu (windows-2022, pytorch, 3.9, 2.1)	16m 9s
	6h 15m 29s

Provide feedback