mytorchsummary


Install
pip install mytorchsummary==1.1