amiriparian
/

ExHuBERT

Audio Classification

Speech Emotion Recognition

Affective Computing

Model card Files Files and versions Community

amiriparian commited on Jun 4

Commit

b158d45

•

1 Parent(s): d995ff8

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -63,6 +63,7 @@ from transformers import AutoModelForAudioClassification, Wav2Vec2FeatureExtract
 model_name = 'amiriparian/HuBERT-EmoSet'
 feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-base-ls960")
 AutoModelForAudioClassification.from_pretrained(model_name, trust_remote_code=True)
 sampling_rate=16000
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 model_name = 'amiriparian/HuBERT-EmoSet'
 feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-base-ls960")
 AutoModelForAudioClassification.from_pretrained(model_name, trust_remote_code=True)
+model.freeze_og_encoder()
 sampling_rate=16000
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")