faiss-gpu==1.7.2
hydra-core==1.3.2
kornia==0.7.1
lightning==2.1.3
matplotlib==3.7.4
scikit-learn==1.3.2
torch==2.1.2
torchmetrics==0.11.4
torchvision==0.16.2
wandb==0.16.0
