Files
vit-pytorch/vit_pytorch/__init__.py
2021-11-12 20:08:48 -08:00

4 lines
98 B
Python

from vit_pytorch.vit import ViT
from vit_pytorch.mae import MAE
from vit_pytorch.dino import Dino