• moto's avatar
    [BC-Breaking] Update `extract_features` of Wav2Vec2Model (#1776) · 78b08c26
    moto authored
    * [BC-Breaking] Update `extract_features` of Wav2Vec2Model
    
    Originally, `extract_features` method was returning the result from
    the convolutional feature extractor module.
    
    The features commonly used in downstream tasks are outputs from intermediate
    layers of transformer block in encoder.
    
    This commit update the behavior of `extract_features` to allow selectively
    retrieve such features.
    78b08c26
model.py 9.48 KB