torch>=2.0
numpy>=1.23
scikit-learn>=1.2
PyYAML>=6.0
tqdm>=4.64
