File size: 314 Bytes
9d06c04
 
 
 
1
2
3
4
from transformers import AutoModelForImageClassification, AutoImageProcessor
m = AutoModelForImageClassification.from_pretrained("dshi01/convnext-tiny-224-7clss")
p = AutoImageProcessor.from_pretrained("dshi01/convnext-tiny-224-7clss")
print(m.config.model_type, m.config.num_labels)  # expect convnext, 7 (if set)