torch>=1.0
torchinfo>=1.7
