• G. Sun's avatar
    Add TCPGen context-biasing Conformer RNN-T (#2890) · 1ed330b5
    G. Sun authored
    Summary:
    This commit adds the implementation of the tree-constrained pointer generator (TCPGen) for contextual biasing.
    
    An example for Librispeech can be found in audio/examples/asr/librispeech_biasing.
    
    Maintainer's note (mthrok):
    It seems that TrieNode should be better typed as tuple, but changing the implementation from list to tuple
    could cause some issue without running the code, so the code is not changed, though the annotation uses tuple.
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/2890
    
    Reviewed By: nateanl
    
    Differential Revision: D43171447
    
    Pulled By: mthrok
    
    fbshipit-source-id: 372bb077d997d720401dbf2dbfa131e6a958e37e
    1ed330b5
__init__.py 1.14 KB