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