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

Model summary in PyTorch similar to `model.summary()` in Keras - sksq96/pytorch-summary
github.com github.com