requirements.txt

-i https://download.pytorch.org/whl/cu118
torch
torchvision
torchaudio
kornia
omegaconf
pytorch-lightning
torchvision
tqdm
torchmetrics
scikit-learn
numpy
matplotlib
git+https://github.com/mhamilton723/CLIP.git
git+https://github.com/mhamilton723/FeatUp