torch>=1.13
torchvision>=0.14
numpy
matplotlib
seaborn
scikit-learn
tqdm
