pytorch-lightning>2,<3
webdataset==0.2.86
easydict==1.13
kornia==0.5.0
hydra-core~=1.3.0
scikit-learn>1.1,<1.6
tensorboard==2.13.0
