amiriparian
commited on
Commit
•
0067bb9
1
Parent(s):
f96b4ac
Update README.md
Browse files
README.md
CHANGED
@@ -64,7 +64,7 @@ model_name = 'amiriparian/ExHuBERT'
|
|
64 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-base-ls960")
|
65 |
model = AutoModelForAudioClassification.from_pretrained(model_name, trust_remote_code=True,revision="b158d45ed8578432468f3ab8d46cbe5974380812")
|
66 |
|
67 |
-
# Freezing half of the encoder
|
68 |
model.freeze_og_encoder()
|
69 |
|
70 |
sampling_rate=16000
|
|
|
64 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-base-ls960")
|
65 |
model = AutoModelForAudioClassification.from_pretrained(model_name, trust_remote_code=True,revision="b158d45ed8578432468f3ab8d46cbe5974380812")
|
66 |
|
67 |
+
# Freezing half of the encoder for further transfer learning
|
68 |
model.freeze_og_encoder()
|
69 |
|
70 |
sampling_rate=16000
|