Model summary in PyTorch similar to `model.summary()` in Keras

WWW: https://github.com/sksq96/pytorch-summary
