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.5.tar.gz (21.2 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.5-py3-none-any.whl (20.7 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: visualtorch-0.2.5.tar.gz
  • Upload date:
  • Size: 21.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.20

File hashes

Hashes for visualtorch-0.2.5.tar.gz
Algorithm Hash digest
SHA256 1b52917e161f15b82e7c737ce6bed11df85bea7a73ffa856cc1b7fb04dd3f70a
MD5 5df66b36febc2bcd99aac6e75be9a439
BLAKE2b-256 8e1568f293508a8c3877354b080f13673284c893c3128e5d7287e1a3e7406427

See more details on using hashes here.

File details

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

File metadata

  • Download URL: visualtorch-0.2.5-py3-none-any.whl
  • Upload date:
  • Size: 20.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.20

File hashes

Hashes for visualtorch-0.2.5-py3-none-any.whl
Algorithm Hash digest
SHA256 665805669e55aee226f816c26637bbf3f6ae5e463b4f980ad56e0bf0595d5edc
MD5 b199d0b84daddb3c23d7465dbf9bc23b
BLAKE2b-256 5ca953ff2a9f9d273949cd042f6cbef68033c14ebb77dad646be060450998a0b

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