numpy~=1.21.5
torch~=1.8.1
torchvision~=0.9.1
matplotlib~=3.4.2
tqdm~=4.61.1