• hwangjeff's avatar
    Refactor LibriSpeech Conformer RNN-T recipe (#2366) · 69467ea5
    hwangjeff authored
    Summary:
    Modifies the example LibriSpeech Conformer RNN-T recipe as follows:
    - Moves data loading and transforms logic from lightning module to data module (improves generalizability and reusability of lightning module and data module).
    - Moves transforms logic from dataloader collator function to dataset (resolves dataloader multiprocessing issues on certain platforms).
    - Replaces lambda functions with `partial` equivalents (resolves pickling issues in certain runtime environments).
    - Modifies training script to allow for specifying path model checkpoint to restart training from.
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/2366
    
    Reviewed By: mthrok
    
    Differential Revision: D36305028
    
    Pulled By: hwangjeff
    
    fbshipit-source-id: 0b768da5d5909136c55418bf0a3c2ddd0c5683ba
    69467ea5
eval.py 2.15 KB