From Pytorch Model Summary Import Summary