/pytorch-summary

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

Primary LanguagePython

Stargazers