README.md
pretrained_model/model.pt
pretrained_model/model.yaml
pretrained_model/config.yaml
models/__init__.py
models/vit/vit.py
models/vit/__init__.py
models/vit/configs/v1_small.yaml
models/vit/configs/v1_base.yaml
models/base/utils.py
models/base/__init__.py
models/base/configs/example.yaml