BirdMAE-XCL / README.md
mwirth7's picture
Update README.md
16a7103 verified
|
Raw
History Blame Contribute Delete
566 Bytes

A supervised finetune of the BirdMAE-Base model on XCL from BirdSet


from transformers import AutoModelForSequenceClassification, AutoFeatureExtractor

fe = AutoFeatureExtractor.from_pretrained("DBD-research-group/BirdMAE-XCL", trust_remote_code=True) # "DBD-research-group/Bird-MAE-Base" also works

model = AutoModelForSequenceClassification.from_pretrained("DBD-research-group/BirdMAE-XCL", trust_remote_code=True, num_labels=9736)

print(model.config.id2label)

# {0: 'ostric2',
#  1: 'grerhe1',
#  2: 'lesrhe2',
#  3: 'sobkiw1',
#  ...
#  }