• Dmytro Okhonko's avatar
    Add ctc loss to ASR task (#1233) · c4893ca6
    Dmytro Okhonko authored
    Summary:
    Adds CTC loss and corresponding transformer ctc based models.
    
    Tested with
    `CUDA_VISIBLE_DEVICES=0 python train.py $DATA_PATH --save-dir $SAVE_DIR --max-epoch 30 --task speech_recognition --arch vggtransformer_enc_1 --optimizer adadelta --lr 1.0 --adadelta-eps 1e-8 --adadelta-rho 0.95 --clip-norm 10.0  --max-tokens 10000 --log-format json --log-interval 1 --criterion ctc_loss --user-dir examples/speech_recognition/ --validate-interval=10`
    Pull Request resolved: https://github.com/pytorch/fairseq/pull/1233
    
    Reviewed By: jcai1
    
    Differential Revision: D17856824
    
    Pulled By: okhonko
    
    fbshipit-source-id: f3eac64d3fdd0c37cf8c539dd360cfb610d8a6ef
    c4893ca6
data_utils.py 3.15 KB