- Linux or macOS with python ≥ 3.6
- PyTorch ≥ 1.6
- torchvision that matches the Pytorch installation. You can install them together at pytorch.org to make sure of this.
- yacs
- Cython (optional to compile evaluation code)
- tensorboard (needed for visualization):
pip install tensorboard - gdown (for automatically downloading pre-train model)
- sklearn
- termcolor
- tabulate
- faiss
pip install faiss-cpu - mat4py
pip install mat4py - pandas