numpy
scikit-learn
torch
torchvision
easy-module-attribute-getter==0.9.41
record-keeper==0.9.29
tensorboard
matplotlib
pretrainedmodels
pytorch-metric-learning==0.9.92
pandas
ax-platform
faiss-gpu
gdown>=3.12.0
