torchinfo

Add model name to the topmost row of the summary table, fixed bug in nested_list_size.

Add row_settings=("var_name", "depth") as customizable options for showing variable names in a row.

Adjust MACs calculation to include batch_size

Last release under the old PyPI package, deprecates Python 3.5 support.

Please pin v1.4.3 if you wish to continue using Python 3.5.
Additionally fixes several misc bugs.

Minor bug fixes, now forwards correct error messages when an exception occurs.

Fix bug with nested list inputs.

Sequential & ModuleList Support

Thanks @roym899 for implementing this feature!

This is likely a one-time release backporting to Python 3.5 by reverting all of the f-strings to format strings.

Minor API changes with device and batch_dim, as well as additional testing on models from issue submitters.

Initial Release