Last updated:
0 purchases
bitorchinfo 0.1.0
bittorchinfo
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.
Usage
pip install torchinfo
Contributing
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
References
Thanks to @tyleryep for providing the torchinfo package.
For personal and professional use. You cannot resell or redistribute these repositories in their original state.
There are no reviews.