• hwangjeff's avatar
    Change underlying implementation of RNN-T hypothesis to tuple (#2339) · 6b242c29
    hwangjeff authored
    Summary:
    PyTorch Lite, which is becoming a standard for mobile PyTorch usage, does not support containers containing custom classes. Consequently, because TorchAudio's RNN-T decoder currently returns and accepts lists of `Hypothesis` namedtuples, it is not compatible with PyTorch Lite. This PR resolves said incompatibility by changing the underlying implementation of `Hypothesis` to tuple.
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/2339
    
    Reviewed By: nateanl
    
    Differential Revision: D35806529
    
    Pulled By: hwangjeff
    
    fbshipit-source-id: 9cbae5504722390511d35e7f9966af2519ccede5
    6b242c29
models.rst 2.86 KB