torch>=1.4.0
torchvision>=0.5.0
numpy>=1.22.3 
scikit-learn>=1.1.1 
pyyaml
