torch==1.13
torchvision
pytorch-lightning~=1.6.5
timm
nltk
lmdb
Pillow
imgaug
hydra-core
fvcore
ray[tune]
ax-platform
PyYAML
tqdm