torch==1.12.0
torchvision==0.13.0
numpy>=1.20.1
scikit-learn>=1.0.2
matplotlib>=3.2.2
kornia>=0.6.9
