GitHub repository: DaunKimY/modelSummary
Description: A custom-renovated version of torchsummary module