Published 7 years ago
Model summary in PyTorch similar to `model.summary()` in Keras
pip install torchsummary
[email protected]
