Aniemore
/

wav2vec2-xlsr-53-russian-emotion-recognition

Audio Classification

feature-extraction

emotion-recognition

emotion-classification

Model card Files Files and versions Community

Ar4ikov commited on May 22, 2022

Commit

8b9f2c3

•

1 Parent(s): 71c53d9

Update wav2vec2speechclassification.py

Files changed (1) hide show

wav2vec2speechclassification.py +3 -3

wav2vec2speechclassification.py CHANGED Viewed

@@ -2,7 +2,7 @@ from dataclasses import dataclass
 from typing import Optional, Tuple
 import torch
 from transformers.file_utils import ModelOutput
-from transformers import AutoConfig
 @dataclass
@@ -25,7 +25,7 @@ from transformers.models.wav2vec2.modeling_wav2vec2 import (
 class Wav2Vec2ClassificationHead(nn.Module):
     """Head for wav2vec classification task."""
-    config_class = AutoConfig
     def __init__(self, config):
         super().__init__()
@@ -44,7 +44,7 @@ class Wav2Vec2ClassificationHead(nn.Module):
 class Wav2Vec2ForSpeechClassification(Wav2Vec2PreTrainedModel):
-    config_class = AutoConfig
     def __init__(self, config):
         super().__init__(config)

 from typing import Optional, Tuple
 import torch
 from transformers.file_utils import ModelOutput
+from transformers import Wav2Vec2Config
 @dataclass
 class Wav2Vec2ClassificationHead(nn.Module):
     """Head for wav2vec classification task."""
+    config_class = Wav2Vec2Config
     def __init__(self, config):
         super().__init__()
 class Wav2Vec2ForSpeechClassification(Wav2Vec2PreTrainedModel):
+    config_class = Wav2Vec2Config
     def __init__(self, config):
         super().__init__(config)