Requirements

Pytorch 1.9.0
torchvision 0.11.0
scipy 1.5.2
cudatoolkit 11.3.1
tensorboardX 2.2
torchvision 0.11.0
Python 3.7
