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