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.
Get Started#
Usage Examples#
API References#
Citation#
Please cite this project in your publications if it helps your research as follows:
@misc{Hendria2024VisualTorch,
author = {Hendria, Willy Fitra},
title = {visualtorch},
year = {2024},
publisher = {GitHub},
journal = {GitHub repository},
note = {\url{https://github.com/willyfh/visualtorch}},
}