Call configure_module before freeze_before_training #26500

Job	Run time
build-packages / init	5s
build-packages / build-packages (lightning)	26s
build-packages / build-packages (fabric)	19s
build-packages / build-packages (pytorch)	24s
install-pkg (ubuntu-22.04, fabric, 3.9)	1m 44s
install-pkg (ubuntu-22.04, fabric, 3.11)	1m 37s
install-pkg (ubuntu-22.04, pytorch, 3.9)	2m 15s
install-pkg (ubuntu-22.04, pytorch, 3.11)	1m 57s
install-pkg (ubuntu-22.04, lightning, 3.9)	1m 59s
install-pkg (ubuntu-22.04, lightning, 3.11)	1m 45s
install-pkg (ubuntu-22.04, notset, 3.9)	1m 50s
install-pkg (ubuntu-22.04, notset, 3.11)	2m 13s
install-pkg (macOS-14, fabric, 3.9)	1m 2s
install-pkg (macOS-14, fabric, 3.11)	31s
install-pkg (macOS-14, pytorch, 3.9)	53s
install-pkg (macOS-14, pytorch, 3.11)	48s
install-pkg (macOS-14, lightning, 3.9)	55s
install-pkg (macOS-14, lightning, 3.11)	39s
install-pkg (macOS-14, notset, 3.9)	51s
install-pkg (macOS-14, notset, 3.11)	55s
install-pkg (windows-2022, fabric, 3.9)	2m 33s
install-pkg (windows-2022, fabric, 3.11)	2m 0s
install-pkg (windows-2022, pytorch, 3.9)	2m 22s
install-pkg (windows-2022, pytorch, 3.11)	2m 39s
install-pkg (windows-2022, lightning, 3.9)	2m 5s
install-pkg (windows-2022, lightning, 3.11)	2m 2s
install-pkg (windows-2022, notset, 3.9)	2m 36s
install-pkg (windows-2022, notset, 3.11)	1m 58s
	41m 23s

Provide feedback