Torchinfo

Latest version: v1.7.2

Safety actively analyzes 625645 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

Page 1 of 5

1.64

What's Changed
* Support half-precision dtypes when calculating model size by jedi2610 in https://github.com/TylerYep/torchinfo/pull/109
* Uninitialized tensor fix by jedi2610 in https://github.com/TylerYep/torchinfo/pull/120
* Allow user to set the training/eval mode of the model before calling summary.

New Contributors
* jedi2610 made their first contribution in https://github.com/TylerYep/torchinfo/pull/109
* bsridatta made their first contribution in https://github.com/TylerYep/torchinfo/pull/116

**Full Changelog**: https://github.com/TylerYep/torchinfo/compare/v1.6.3...v1.64

1.7.2

What's Changed
* Simplify layers_to_str by mert-kurttutan in https://github.com/TylerYep/torchinfo/pull/173
* Add hide_recursive_layers option to ColumnSettings by mert-kurttutan in https://github.com/TylerYep/torchinfo/pull/174
* Add contribution of recursive calls to total_output_bytes by mert-kurttutan in https://github.com/TylerYep/torchinfo/pull/181
* Add params and MACs units specifiers by richardtml in https://github.com/TylerYep/torchinfo/pull/188
* Add the column 'Params %' by luke396 in https://github.com/TylerYep/torchinfo/pull/199
* Automatically get device from first model parameter if `device` parameter to `summary` is `None` by snimu in https://github.com/TylerYep/torchinfo/pull/211
* Enable analyzing nested input- and output-dicts by snimu in https://github.com/TylerYep/torchinfo/pull/212

Cleanup Work
* Add additional versions to GitHub Actions tests by TylerYep in https://github.com/TylerYep/torchinfo/pull/189
* Fix torchvision deprecation warnings in test cases by TylerYep in https://github.com/TylerYep/torchinfo/pull/191
* Separate nested_list_size function, add some documentation, improve mypy for setuptools by TylerYep in https://github.com/TylerYep/torchinfo/pull/220

New Contributors
* richardtml made their first contribution in https://github.com/TylerYep/torchinfo/pull/188
* luke396 made their first contribution in https://github.com/TylerYep/torchinfo/pull/199
* snimu made their first contribution in https://github.com/TylerYep/torchinfo/pull/211

**Full Changelog**: https://github.com/TylerYep/torchinfo/compare/v1.7.1...v1.7.2

1.7.1

What's Changed
* Update half precision test cases to support Pytorch v1.12 by mert-kurttutan in https://github.com/TylerYep/torchinfo/pull/165
* Use layer_id instead of class_name in add_missing_layers by mert-kurttutan in https://github.com/TylerYep/torchinfo/pull/163
* Replace add_missing_layers with add_missing_container_layers by mert-kurttutan in https://github.com/TylerYep/torchinfo/pull/169

New Contributors
* mert-kurttutan made their first contribution in https://github.com/TylerYep/torchinfo/pull/165

**Full Changelog**: https://github.com/TylerYep/torchinfo/compare/v1.7.0...v1.7.1

1.7.0

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**: https://github.com/TylerYep/torchinfo/compare/v1.6.6...v1.7.0

1.6.6

What's Changed
* Add "Trainable" column by bsridatta in https://github.com/TylerYep/torchinfo/pull/128
* Do not error if there are None values in ModuleLists
* Obtain kernel_size using the actual attribute name.

1.6.5

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

Page 1 of 5

© 2024 Safety CLI Cybersecurity Inc. All Rights Reserved.