numpy==1.21.4
torchinfo==1.8.0