https://github.com/facebookresearch/pythia
Raw File
Tip revision: c7cac1b65230b806733e97f1e87393af4a3e60f4 authored by Amanpreet Singh on 29 April 2021, 23:00:03 UTC
[fix] CircleCI GPU tests; sharded DDP tests
Tip revision: c7cac1b
requirements.txt
torch>=1.6.0, <=1.8.1
torchvision>=0.7.0, <=0.9.1
numpy>=1.16.6
tqdm>=4.43.0,<4.50.0
demjson==2.2.4
torchtext==0.5.0
GitPython==3.1.0
requests==2.23.0
fasttext==0.9.1
nltk==3.4.5
editdistance==0.5.3
transformers==3.4.0
sklearn==0.0
omegaconf==2.0.6
lmdb==0.98
termcolor==1.1.0
iopath==0.1.8
datasets==1.2.1
matplotlib==3.3.4
pycocotools==2.0.2
ftfy==5.8
psutil
pytorch-lightning @ git+https://github.com/PyTorchLightning/pytorch-lightning@7b283e3c
torchaudio>=0.6.0, <=0.8.1
back to top