    def get_normalized_probs(
        self,
        net_output,
        log_probs,
        sample=None,
    ):
        """Get normalized probabilities (or log probs) from a net's output."""
        if isinstance(net_output, dict):
            return self.speech_decoder.get_normalized_probs(
                net_output, log_probs, sample
            )
        return self.text_decoder.get_normalized_probs(net_output, log_probs, sample)
Get normalized probabilities (or log probs) from a net's output.
examples/speech_text_joint_to_text/models/joint_speech_text_pretrain_transformer.py