torch>=1.7.0
torchinfo>=1.7.0
