torch>=1.5.0
torchinfo>=0.0.9
torchvision>=0.6
