def on_train_epoch_start(self) -> None:
        """Display a message when starting to train the upsampling module."""
        if self.current_epoch == self.second_phase:
            logger.info("Now training upsampling module.")
Display a message when starting to train the upsampling module.
src/anomalib/models/image/dsr/lightning_model.py