Skip to main content

Architecture visualization of Torch models

Project description

🔥 VisualTorch 🔥

python pytorch Downloads Run Tests Documentation Status

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, pytorchviz, and pytorch-summary.

Note: VisualTorch may not yet support complex models, but contributions are welcome!

VisualTorch Examples

Documentation

Online documentation is available at visualtorch.readthedocs.io.

The docs include usage examples, API references, and other useful information.

Installation

See the Installation page.

Examples

See the Usage Examples page.

Contributing

Please feel free to send a pull request to contribute to this project by following this guideline.

License

This poject is available as open source under the terms of the MIT License.

Originally, this project was based on the visualkeras (under the MIT license), with additional modifications inspired by pytorchviz, and 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 is available.

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

visualtorch-0.2.4.tar.gz (17.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

visualtorch-0.2.4-py3-none-any.whl (20.0 kB view details)

Uploaded Python 3

File details

Details for the file visualtorch-0.2.4.tar.gz.

File metadata

  • Download URL: visualtorch-0.2.4.tar.gz
  • Upload date:
  • Size: 17.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.10.17

File hashes

Hashes for visualtorch-0.2.4.tar.gz
Algorithm Hash digest
SHA256 662e32040533cbbf389cfd12e0a95799f0cf16612bac5eea149e6028bbb36812
MD5 bafebe347549303b7f1e352d19db1d11
BLAKE2b-256 2d092c42a3d9c531ee794f235d6eb0a8e49d2b92adf7c6009932349913b2dfb1

See more details on using hashes here.

File details

Details for the file visualtorch-0.2.4-py3-none-any.whl.

File metadata

  • Download URL: visualtorch-0.2.4-py3-none-any.whl
  • Upload date:
  • Size: 20.0 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.10.17

File hashes

Hashes for visualtorch-0.2.4-py3-none-any.whl
Algorithm Hash digest
SHA256 b84e9115204d3a4d1c2d054f85cdcc7ae0e154ece18340622e6ea832736c442c
MD5 26624c067a8746c7fd644af436a4d94f
BLAKE2b-256 3a0f504fcec35cc8eaf959364482e1dc2b061b279f49fbd0bab0d4df2742ddee

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page