• hwangjeff's avatar
    Clean up Emformer (#2207) · 87d7694d
    hwangjeff authored
    Summary:
    - Make `segment_length` a required argument rather than optional argument to force users to consciously choose input segment lengths for their use cases.
    - Clarify expected input shapes in API documentation.
    - Adjust `infer` tests to reflect expected usage.
    - Add assertion for input shape for `infer`.
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/2207
    
    Reviewed By: mthrok
    
    Differential Revision: D34101205
    
    Pulled By: hwangjeff
    
    fbshipit-source-id: 1d1233d5edee5818d4669b4e47d44559e7ebb304
    87d7694d
rnnt.py 33.6 KB