Make sure that you have a working git, Python 3 and poetry installation before proceeding.
- Clone this repository:
git clone [email protected]:marios1861/T-TAME.git
- Go to the locally saved repository path:
cd T-TAME
- Install:
poetry install
Note: You may need to modify the venv activate script in the case that cuda is already installed in your machine. If so, add this line:
export LD_LIBRARY_PATH=.../venv/lib/python3.8/site-packages/nvidia/cublas/lib/:$LD_LIBRARY_PATH
You can evaluate the attention mechanism trained on vit on the 7th epoch using the command:
tame val ---cfg vit_b_16.yaml epoch 7