| 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 | |