torch
torchvision
tqdm

[dev]
pytorch
torchvision
matplotlib
seaborn
numpy
tqdm
