• Will Rice's avatar
    Adding TFWav2Vec2Model (#11617) · d438eee0
    Will Rice authored
    
    
    * [WIP] Add TFWav2Vec2Model
    
    Work in progress for adding a tensorflow version of Wav2Vec2
    
    * feedback changes
    
    * small fix
    
    * Test Feedback Round 1
    
    * Add SpecAugment and CTC Loss
    
    * correct spec augment mask creation
    
    * docstring and correct copyright
    
    * correct bugs
    
    * remove bogus file
    
    * finish tests correction
    
    * del unnecessary layers
    
    * Update src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    
    * make style
    
    * correct final bug
    
    * Feedback Changes
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    d438eee0
test_modeling_tf_common.py 62.7 KB