https://github.com/facebookresearch/pythia
Raw File
Tip revision: da297e47d7612edf78afd21b03fa6b1af7ca7b39 authored by ryan-qiyu-jiang on 13 December 2021, 21:11:52 UTC
Update base for Update on "[docs] Add ViLT model checkpoint instructions"
Tip revision: da297e4
requirements.txt
torch>=1.6.0, <=1.9.0
torchvision>=0.7.0, <=0.10.0
numpy>=1.16.6
tqdm>=4.43.0,<4.50.0
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, <=4.10.1
sklearn==0.0
omegaconf>=2.0.6, <=2.1
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
pytorch-lightning @ git+https://github.com/PyTorchLightning/pytorch-lightning@5b9995da0
torchaudio>=0.6.0, <=0.9.0
psutil
pillow==8.3.1
back to top