Skip to main content

Model summary for bitorch, based on the original torchinfo.

Project description


This project is an extended version of torchinfo. However we added support for binary layers, as used in bitorch. Please check out and use the original project, unless you specifically want to use this package in conjunction with bitorch.


pip install torchinfo


All issues and pull requests are much appreciated! If you are wondering how to build the project:

  • torchinfo is actively developed using the lastest version of Python.
    • Changes should be backward compatible with Python 3.6, but this is subject to change in the future.
    • Run pip install -r requirements-dev.txt. We use the latest versions of all dev packages.
    • Run pre-commit install.
    • To use auto-formatting tools, use pre-commit run -a.
    • To run unit tests, run pytest.
    • To update the expected output files, run pytest --overwrite.
    • To skip output file tests, use pytest --no-output


  • Thanks to @tyleryep for providing the torchinfo package.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

bitorchinfo-0.1.0.tar.gz (14.0 kB view hashes)

Uploaded source

Built Distribution

bitorchinfo-0.1.0-py3-none-any.whl (14.6 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page