requirements.txt

appdirs==1.4.4
certifi==2023.5.7
charset-normalizer==3.1.0
click==8.1.3
docker-pycreds==0.4.0
einops==0.6.1
filelock==3.12.2
fsspec==2023.6.0
fvcore==0.1.5.post20221221
gitdb==4.0.10
GitPython==3.1.31
huggingface-hub==0.16.4
idna==3.4
iopath==0.1.10
numpy==1.24.3
nvidia-cublas-cu11==11.10.3.66
nvidia-cuda-nvrtc-cu11==11.7.99
nvidia-cuda-runtime-cu11==11.7.99
nvidia-cudnn-cu11==8.5.0.96
packaging==23.1
pandas==2.1.0
pathtools==0.1.2
Pillow==9.5.0
portalocker==2.8.2
protobuf==4.23.2
psutil==5.9.5
python-dateutil==2.8.2
pytz==2023.3.post1
PyYAML==6.0
requests==2.31.0
safetensors==0.3.2
scipy==1.10.1
sentry-sdk==1.25.1
setproctitle==1.3.2
six==1.16.0
smmap==5.0.0
tabulate==0.9.0
termcolor==2.3.0
timm==0.9.5
torch==1.13.1
torchaudio==0.13.1
torchsummary==1.5.1
torchvision==0.14.1
tqdm==4.66.0
typing_extensions==4.6.3
tzdata==2023.3
urllib3==2.0.3
wandb==0.15.4
warmup-scheduler==0.3
yacs==0.1.8