torch-summary

Model summary in PyTorch, based off of the original torchsummary.


Keywords
torch, pytorch, torchsummary, torch-summary, summary, keras, deep-learning, ml, torchinfo, torch-info, visualize, model, statistics, layer
License
MIT
Install
pip install torch-summary==1.4.5