GitHub repository: m3at/pytorch-summary
Description: Model summary in PyTorch similar to `model.summary()` in Keras