LICENSE
README.md
pyproject.toml
tests/test_data.py
tests/test_models.py
tests/test_trainer.py
vit_trainer/__init__.py
vit_trainer/cli.py
vit_trainer/config.py
vit_trainer.egg-info/PKG-INFO
vit_trainer.egg-info/SOURCES.txt
vit_trainer.egg-info/dependency_links.txt
vit_trainer.egg-info/entry_points.txt
vit_trainer.egg-info/requires.txt
vit_trainer.egg-info/top_level.txt
vit_trainer/data/__init__.py
vit_trainer/data/cifar.py
vit_trainer/data/transforms.py
vit_trainer/evaluation/__init__.py
vit_trainer/evaluation/metrics.py
vit_trainer/models/__init__.py
vit_trainer/models/vit.py
vit_trainer/training/__init__.py
vit_trainer/training/callbacks.py
vit_trainer/training/trainer.py
vit_trainer/visualization/__init__.py
vit_trainer/visualization/attention.py