matplotlib
numpy
pandas
scikit-learn
torch==1.8.2
torchaudio
torchvision
tqdm