Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
So, to use that on your own image classification dataset,
use the [ViTMSNForImageClassification] class which is initialized from [ViTMSNModel].