What's Changed

• Enable np.ndarrays as in- and outputs. by @snimu in #222
• Don't change device if input_data is given by @snimu in #236
• Move to using tensor.untyped_storage() for Pytorch 2.0 by @TylerYep in 7f2bed3

Full Changelog: v1.7.2...v1.8.0

What's Changed

• Simplify layers_to_str by @mert-kurttutan in #173
• Add hide_recursive_layers option to ColumnSettings by @mert-kurttutan in #174
• Add contribution of recursive calls to total_output_bytes by @mert-kurttutan in #181
• Add params and MACs units specifiers by @richardtml in #188
• Add the column 'Params %' by @luke396 in #199
• Automatically get device from first model parameter if device parameter to summary is None by @snimu in #211
• Enable analyzing nested input- and output-dicts by @snimu in #212

Cleanup Work

• Add additional versions to GitHub Actions tests by @TylerYep in #189
• Fix torchvision deprecation warnings in test cases by @TylerYep in #191
• Separate nested_list_size function, add some documentation, improve mypy for setuptools by @TylerYep in #220

New Contributors

• @richardtml made their first contribution in #188
• @luke396 made their first contribution in #199
• @snimu made their first contribution in #211

Full Changelog: v1.7.1...v1.7.2

What's Changed

• Update half precision test cases to support Pytorch v1.12 by @mert-kurttutan in #165
• Use layer_id instead of class_name in add_missing_layers by @mert-kurttutan in #163
• Replace add_missing_layers with add_missing_container_layers by @mert-kurttutan in #169

New Contributors

• @mert-kurttutan made their first contribution in #165

Full Changelog: v1.7.0...v1.7.1

What's Changed

• Calculate parameter counts for nn.Parameter
• Add parameter counts for nn.UninitializedParameters
• Show output shapes for the overall model
• Internal optimizations (hooks applied iteratively instead of recursively, add profiling code)

Full Changelog: v1.6.6...v1.7.0

What's Changed

• Add "Trainable" column by @bsridatta in #128
• Do not error if there are None values in ModuleLists
• Obtain kernel_size using the actual attribute name.

Fix a regression in which torchinfo would crash on PyTorch versions < 1.9

Fixed bug with layer reuse in the same variable, with and without existing hooks.

• Support for pruned models
• Support for "ascii_only" as a row setting to disable the fancy branch logging.

Deprecate Python 3.6 support. Install v1.5.4 if you want to use Python 3.6.