torch==2.5.1
torchvision==0.20.1
numpy==2.1.2
scikit-learn==1.5.2
