numpy
matplotlib
tqdm==4.56.0
torch>=1.7.1
torchvision>=0.8.2
