search

b04901014 / FT-w2v2-ser

Official implementation for the paper Exploring Wav2vec 2.0 fine-tuning for improved speech emotion recognition
MIT License
136 stars 32 forks source link

TypeError: _compute_mask_indices() got an unexpected keyword argument 'device' #1

Closed ja5p3r closed 2 years ago

ja5p3r commented 2 years ago

Hi while running run_downstream_custom_multiple_fold.py, I ran into TypeError: _compute_mask_indices() got an unexpected keyword argument 'device' In package transformers modeling_wav2vec2.py also shows that _compute_mask_indices() does not take arg device Am I missing something here?

Any help is appreciated!

b04901014 commented 2 years ago

I think you installed the latest version 4.12.0 released recently where they changed the function arguments of _compute_mask_indices(). We haven't update that correspondingly and the code is compatible for version 4.10.0. You can view the docker/Dockerfile for the specific version of packages that is compatible when the project is finished. Or you can just try to just delete the device argument.

_compute_mask_indices() of before 4.11.3: https://huggingface.co/transformers/v4.11.3/_modules/transformers/models/wav2vec2/modeling_wav2vec2.html#Wav2Vec2Model

ja5p3r commented 2 years ago

Understood Thanks!