• Zhaoheng Ni's avatar
    Fix hubert fine-tuning recipe (#2851) · 980528e9
    Zhaoheng Ni authored
    Summary:
    - `_get_fileids_paths` in `LibriLightLimited` dataset was changed dataset in https://github.com/pytorch/audio/issues/2653, the absolute path becomes relative paths. This PR fixes the usage in hubert fine-tuning recipe to get correct audio paths.
    - model options should be `hubert_pretrain_large` and `hubert_pretrain_xlarge` instead of `hubert_large` and `hubert_xlarge`.
    - The input dimension of CTC linear layer varies depending on the model architecture, update it in lightning module.
    
    cc simpleoier
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/2851
    
    Reviewed By: carolineechen
    
    Differential Revision: D41327998
    
    Pulled By: nateanl
    
    fbshipit-source-id: f92248ee84ec860b4e4dbef880c5794b338e1e2d
    980528e9
hubert_dataset.py 20.4 KB