numpy
torch
torchvision
faiss-cpu
timm
scikit-learn
tqdm
