torch==1.6.0
torchvision==0.7.0
numpy==1.18.5
imageio==2.9.0
matplotlib==3.2.2