<div align="center">
<h1>🔥 VisualTorch 🔥</h1>
[]() []() [](https://pepy.tech/project/visualtorch) [](https://github.com/willyfh/visualtorch/actions/workflows/pytest.yml) [](https://visualtorch.readthedocs.io/en/latest/?badge=latest)
</div>
**VisualTorch** aims to help visualize Torch-based neural network architectures. It currently supports generating layered-style, graph-style, and LeNet-style architectures for PyTorch Sequential and Custom models. This tool is adapted from [visualkeras](https://github.com/paulgavrikov/visualkeras), [pytorchviz](https://github.com/szagoruyko/pytorchviz), and [pytorch-summary](https://github.com/sksq96/pytorch-summary).
**Note:** VisualTorch may not yet support complex models, but contributions are welcome!
<div align="center">
</div>
## Documentation
Online documentation is available at [visualtorch.readthedocs.io](https://visualtorch.readthedocs.io/en/latest/).
The docs include [usage examples](https://visualtorch.readthedocs.io/en/latest/usage_examples/index.html), [API references](https://visualtorch.readthedocs.io/en/latest/markdown/api_references/index.html), and other useful information.
## Installation
See the [Installation page](https://visualtorch.readthedocs.io/en/latest/markdown/get_started/installation.html).
## Examples
See the [Usage Examples page](https://visualtorch.readthedocs.io/en/latest/usage_examples/index.html).
## Contributing
Please feel free to send a pull request to contribute to this project.
## License
This poject is available as open source under the terms of the [MIT License](https://github.com/willyfh/visualtorch/blob/update-readme/LICENSE).
Originally, this project was based on the [visualkeras](https://github.com/paulgavrikov/visualkeras) (under the MIT license), with additional modifications inspired by [pytorchviz](https://github.com/szagoruyko/pytorchviz), and [pytorch-summary](https://github.com/sksq96/pytorch-summary), both of which are also licensed under the MIT license.
## Citation
Please cite this project in your publications if it helps your research.
[A ready-made citation entry](https://visualtorch.readthedocs.io/en/latest/index.html#citation) is available.
Raw data
{
"_id": null,
"home_page": "https://github.com/willyfh/visualtorch",
"name": "visualtorch",
"maintainer": null,
"docs_url": null,
"requires_python": ">=3.10",
"maintainer_email": null,
"keywords": "visualize architecture, torch visualization, visualtorch",
"author": "Willy Fitra Hendria",
"author_email": "willyfitrahendria@gmail.com",
"download_url": "https://files.pythonhosted.org/packages/c4/c4/e1b0109e2125b45b9a1f49207fa9c81ed69859b51db33265417f044f16ac/visualtorch-0.2.2.tar.gz",
"platform": null,
"description": "<div align=\"center\">\n <h1>\ud83d\udd25 VisualTorch \ud83d\udd25</h1>\n\n[]() []() [](https://pepy.tech/project/visualtorch) [](https://github.com/willyfh/visualtorch/actions/workflows/pytest.yml) [](https://visualtorch.readthedocs.io/en/latest/?badge=latest)\n\n</div>\n\n**VisualTorch** aims to help visualize Torch-based neural network architectures. It currently supports generating layered-style, graph-style, and LeNet-style architectures for PyTorch Sequential and Custom models. This tool is adapted from [visualkeras](https://github.com/paulgavrikov/visualkeras), [pytorchviz](https://github.com/szagoruyko/pytorchviz), and [pytorch-summary](https://github.com/sksq96/pytorch-summary).\n\n**Note:** VisualTorch may not yet support complex models, but contributions are welcome!\n\n<div align=\"center\">\n\n\n\n</div>\n\n## Documentation\n\nOnline documentation is available at [visualtorch.readthedocs.io](https://visualtorch.readthedocs.io/en/latest/).\n\nThe docs include [usage examples](https://visualtorch.readthedocs.io/en/latest/usage_examples/index.html), [API references](https://visualtorch.readthedocs.io/en/latest/markdown/api_references/index.html), and other useful information.\n\n## Installation\n\nSee the [Installation page](https://visualtorch.readthedocs.io/en/latest/markdown/get_started/installation.html).\n\n## Examples\n\nSee the [Usage Examples page](https://visualtorch.readthedocs.io/en/latest/usage_examples/index.html).\n\n## Contributing\n\nPlease feel free to send a pull request to contribute to this project.\n\n## License\n\nThis poject is available as open source under the terms of the [MIT License](https://github.com/willyfh/visualtorch/blob/update-readme/LICENSE).\n\nOriginally, this project was based on the [visualkeras](https://github.com/paulgavrikov/visualkeras) (under the MIT license), with additional modifications inspired by [pytorchviz](https://github.com/szagoruyko/pytorchviz), and [pytorch-summary](https://github.com/sksq96/pytorch-summary), both of which are also licensed under the MIT license.\n\n## Citation\n\nPlease cite this project in your publications if it helps your research.\n\n[A ready-made citation entry](https://visualtorch.readthedocs.io/en/latest/index.html#citation) is available.\n",
"bugtrack_url": null,
"license": "MIT",
"summary": "Architecture visualization of Torch models",
"version": "0.2.2",
"project_urls": {
"Homepage": "https://github.com/willyfh/visualtorch"
},
"split_keywords": [
"visualize architecture",
" torch visualization",
" visualtorch"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "67c2ca6fe6c4187dbc9bbb56a04ac9c9c9db938d319a0fc401a3e656d7abfbf0",
"md5": "cff022e3df039e849f65599a4dc50e76",
"sha256": "16f81858309e0954e84e8e692bd63e96cb2d05c51f8a6f11234fc40a366267af"
},
"downloads": -1,
"filename": "visualtorch-0.2.2-py3-none-any.whl",
"has_sig": false,
"md5_digest": "cff022e3df039e849f65599a4dc50e76",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.10",
"size": 19719,
"upload_time": "2024-04-10T15:15:25",
"upload_time_iso_8601": "2024-04-10T15:15:25.952530Z",
"url": "https://files.pythonhosted.org/packages/67/c2/ca6fe6c4187dbc9bbb56a04ac9c9c9db938d319a0fc401a3e656d7abfbf0/visualtorch-0.2.2-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "c4c4e1b0109e2125b45b9a1f49207fa9c81ed69859b51db33265417f044f16ac",
"md5": "d52c2aca804e2bd5a0b05e6fe9e30104",
"sha256": "72f47a35cd632be44765a77c7a55a3822eeb24d56a7d2928e548da8b824c826e"
},
"downloads": -1,
"filename": "visualtorch-0.2.2.tar.gz",
"has_sig": false,
"md5_digest": "d52c2aca804e2bd5a0b05e6fe9e30104",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.10",
"size": 17186,
"upload_time": "2024-04-10T15:15:27",
"upload_time_iso_8601": "2024-04-10T15:15:27.618018Z",
"url": "https://files.pythonhosted.org/packages/c4/c4/e1b0109e2125b45b9a1f49207fa9c81ed69859b51db33265417f044f16ac/visualtorch-0.2.2.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-04-10 15:15:27",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "willyfh",
"github_project": "visualtorch",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"lcname": "visualtorch"
}
JSON
