flax-community
/

hubert-dementia-screening

Feature Extraction

Model card Files Files and versions

patrickvonplaten commited on Jul 6, 2021

Commit

47a6a98

·

1 Parent(s): 004e84f

Update run_hubert_classifier.py

Files changed (1) hide show

run_hubert_classifier.py +1 -1

run_hubert_classifier.py CHANGED Viewed

@@ -2,7 +2,7 @@
 from hubert_for_sequence_classification import FlaxHubertForSequenceClassification, FlaxHubertModel
 import numpy as np
-# need to do some ugly save/reload because of a bug
 model = FlaxHubertModel.from_pretrained("facebook/hubert-large-ll60k", from_pt=True)
 model.save_pretrained("./")
 model = FlaxHubertForSequenceClassification.from_pretrained("./")

 from hubert_for_sequence_classification import FlaxHubertForSequenceClassification, FlaxHubertModel
 import numpy as np
+# need to do some ugly save/reload because of a bug: https://github.com/huggingface/transformers/issues/12532
 model = FlaxHubertModel.from_pretrained("facebook/hubert-large-ll60k", from_pt=True)
 model.save_pretrained("./")
 model = FlaxHubertForSequenceClassification.from_pretrained("./")