numpy
scikit-learn
tqdm
torch>=1.6.0
torchvision

[with-hooks]
record-keeper>=0.9.30
faiss-gpu>=1.6.3
tensorboard

[with-hooks-cpu]
record-keeper>=0.9.30
faiss-cpu>=1.6.3
tensorboard
