torchsummary is a Python module that produces model summary in PyTorch similar
to `model.summary()` in Keras.
