Update README.md
Browse files
README.md
CHANGED
|
@@ -63,6 +63,7 @@ from transformers import AutoModelForAudioClassification, Wav2Vec2FeatureExtract
|
|
| 63 |
model_name = 'amiriparian/HuBERT-EmoSet'
|
| 64 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-base-ls960")
|
| 65 |
AutoModelForAudioClassification.from_pretrained(model_name, trust_remote_code=True)
|
|
|
|
| 66 |
|
| 67 |
sampling_rate=16000
|
| 68 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 63 |
model_name = 'amiriparian/HuBERT-EmoSet'
|
| 64 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-base-ls960")
|
| 65 |
AutoModelForAudioClassification.from_pretrained(model_name, trust_remote_code=True)
|
| 66 |
+
model.freeze_og_encoder()
|
| 67 |
|
| 68 |
sampling_rate=16000
|
| 69 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|