numpy==1.20.2
scipy==1.6.3
tensorboard==2.5.0
torch==1.8.1
torchvision==0.9.1