matplotlib
numpy
pandas
Pillow
scipy
six
tqdm
tensorboard==2.3.0
torch
torchvision