requirements.txt

attrs==22.1.0
certifi @ file:///opt/conda/conda-bld/certifi_1655968806487/work/certifi
charset-normalizer==2.1.0
click==8.1.3
coolname==1.1.0
docker-pycreds==0.4.0
docstring-parser==0.14.1
einops==0.4.1
fsspec==2022.7.1
gitdb==4.0.9
GitPython==3.1.27
idna==3.3
jsonschema==4.7.2
mosaicml==0.8.2
numpy==1.23.1
packaging==21.3
pathtools==0.1.2
Pillow-SIMD==9.0.0.post1
portalocker==2.5.1
promise==2.3
protobuf==3.20.1
psutil==5.9.1
py-cpuinfo==8.0.0
pyarrow==9.0.0
pyDeprecate==0.3.2
pyparsing==3.0.9
pyrsistent==0.18.1
pytimeparse==1.1.8
pytorch-ranger==0.1.1
PyYAML==6.0
repackage==0.7.3
requests==2.28.1
ruamel.yaml==0.17.21
ruamel.yaml.clib==0.2.6
sentry-sdk==1.9.2
setproctitle==1.3.1
shortuuid==1.0.9
six==1.16.0
smmap==5.0.0
torch==1.12.1+cu116
torch-optimizer==0.1.0
torchdata==0.4.1
torchmetrics==0.7.3
torchvision==0.13.1+cu116
tqdm==4.64.0
typer==0.6.1
typing_extensions==4.3.0
urllib3==1.26.11
wandb==0.12.21
yahp==0.1.3