Isa0
/

text-to-speech-train-code

Isa0 commited on Jun 8

Commit

389df6d

1 Parent(s): d353c94

fix: change model.freeze_feature_extractor() to model.freeze_feature_encoder()

Files changed (1) hide show

train.py CHANGED Viewed

@@ -102,7 +102,7 @@ if __name__ == "__main__":
         pad_token_id=processor.tokenizer.pad_token_id,
         vocab_size=len(processor.tokenizer),
     )
-    model.freeze_feature_extractor()
     training_args = TrainingArguments(
         output_dir="wav2vec2-ljspeech",

         pad_token_id=processor.tokenizer.pad_token_id,
         vocab_size=len(processor.tokenizer),
     )
+    model.freeze_feature_encoder()
     training_args = TrainingArguments(
         output_dir="wav2vec2-ljspeech",