Python GreedyRNNTInferConfig Exemples

Exemple #1

Fichier : rnnt_wer.py Projet : askaydevs/ITN_Phore

class RNNTDecodingConfig:
    strategy: str = "greedy_batch"
    compute_hypothesis_token_set: bool = False

    # greedy decoding config
    greedy: greedy_decode.GreedyRNNTInferConfig = greedy_decode.GreedyRNNTInferConfig(
    )

    # beam decoding config
    beam: beam_decode.BeamRNNTInferConfig = beam_decode.BeamRNNTInferConfig(
        beam_size=4)

Exemple #2

Fichier : rnnt_wer.py Projet : quuhua911/NeMo

class RNNTDecodingConfig:
    strategy: str = "greedy_batch"
    compute_hypothesis_token_set: bool = False

    # preserve decoding alignments
    preserve_alignments: Optional[bool] = None

    # RNNT Joint fused batch size
    fused_batch_size: Optional[int] = None

    # greedy decoding config
    greedy: greedy_decode.GreedyRNNTInferConfig = greedy_decode.GreedyRNNTInferConfig(
    )

    # beam decoding config
    beam: beam_decode.BeamRNNTInferConfig = beam_decode.BeamRNNTInferConfig(
        beam_size=4)