/DistillableVisionTransformer

Easy usage of distillation tokens with vision transformers. Includes models pretrained on ImageNet.

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers