• hwangjeff's avatar
    Apply minor fixes to Emformer implementation (#2252) · cbf1b839
    hwangjeff authored
    Summary:
    Noticed some items to clean up in `Emformer`.
    - Make `segment_length` a required argument in `_EmformerLayer`.
    - Remove unused variables from `_unpack_state` and `_gen_attention_mask`.
    
    These don't affect `Emformer`'s functionality or public API.
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/2252
    
    Reviewed By: carolineechen, mthrok
    
    Differential Revision: D34321430
    
    Pulled By: hwangjeff
    
    fbshipit-source-id: 38a5046f633a3e625352c476ef71c78380ccc597
    cbf1b839
emformer.py 36 KB