torch>=2.3.0
torchvision>=0.15.2
numpy>=1.26.4
tqdm<5,>=4.66.2
pyyaml>=6.0.1
scikit-learn<2,>=1.4.2
scipy<2,>=1.13.0
joblib<2,>=1.4.0
