requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu113
accimage==0.2.0
colorama==0.4.5
einops==0.4.1
fvcore==0.1.5.post20220512
matplotlib==3.5.1
numpy==1.21.5
pandas==1.3.5
Pillow==9.0.0
progress==1.6
tensorboardX==2.6
timm==0.4.12
torch==1.12.1+cu113
torchvision==0.13.1+cu113
lmdb==1.4.1
tensorboard==2.11.2
accelerate==0.20.3
jupyter==1.0.0
notebook==6.4.12
gdown==4.7.3