Skip to content

Initial release

Latest

Choose a tag to compare

@eoffermann eoffermann released this 21 Dec 17:40
· 6 commits to main since this release

Release v0.1.0 - Initial Release

TorchModelAnalyzer v0.1.0 is the first release of this utility for analyzing PyTorch .pth model files. This tool provides a detailed JSON report about the structure, keys, and parameters stored in .pth files, including:

  • Identification of keys like state_dict, optimizer, params, and params_ema.
  • Analysis of parameter counts and shapes.
  • High-level summaries of file contents.
  • Module hierarchy and detailed layer information for saved models.

This release is ideal for developers and researchers who need insights into their PyTorch model files without manually inspecting them. Future updates will enhance functionality based on community feedback.