Model summary in PyTorch similar to `model.summary()` in Keras - View it on GitHub