dshi01
/

convnext-tiny-224-7clss

Model card Files Files and versions

convnext-tiny-224-7clss / testmodel.py

danielhshi8224

fix model nesting

aa94099 4 months ago

history blame contribute delete

356 Bytes

	from transformers import AutoModelForImageClassification, AutoImageProcessor
	m = AutoModelForImageClassification.from_pretrained("dshi01/my-benthic-classifier/convnext-tiny-224-7cls")
	p = AutoImageProcessor.from_pretrained("dshi01/my-benthic-classifier/convnext-tiny-224-7cls")
	print(m.config.model_type, m.config.num_labels) # expect convnext, 7 (if set)