torch>=1.10.0
torchvision>=0.11.0
numpy
tqdm
matplotlib
seaborn
