VisualTorch Documentation

VisualTorch Documentation#

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.

VisualTorch

Get Started#

Installation

Installing visualtorch via PyPI or Source.

Installation

Usage Examples#

All Examples

Usage examples of VisualTorch.

Usage Examples

API References#

All API References

API references of VisualTorch.

API References

Developer Guides#

Contributing to VisualTorch

How to contribute, prepare the development environment, test, report issues, and seek support.

Contributing to VisualTorch

Citation#

Please cite this project in your publications if it helps your research as follows:

@article{Hendria2024,
  doi = {10.21105/joss.06678},
  url = {https://doi.org/10.21105/joss.06678},
  year = {2024}, publisher = {The Open Journal},
  volume = {9},
  number = {102},
  pages = {6678},
  author = {Willy Fitra Hendria and Paul Gavrikov},
  title = {VisualTorch: Streamlining Visualization for PyTorch Neural Network Architectures},
  journal = {Journal of Open Source Software}
}
Contents