558 B

Raw Blame History

Installation

Requirements

Linux or macOS with python ≥ 3.6
PyTorch ≥ 1.2
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