可视化你的
PyTorch 神经网络

TorchNetViz 提供直观的模型结构可视化，帮助深度学习研究者和开发者更好地理解、调试和分享他们的神经网络架构。

创建或加载你的神经网络模型

使用一行代码生成模型可视化

在浏览器中交互式查看模型结构

强大的可视化工具

TorchNetViz 提供了丰富的功能，帮助你更好地理解和分析深度学习模型

生成详细的模型结构图，支持层间连接可视化、模块嵌套显示和交互式探索。

自动推断并显示各层输入输出张量形状，帮助识别维度不匹配问题和理解数据流动。

支持PyTorch内置模块和自定义模型，无缝集成到现有开发流程中。

支持导出HTML，方便嵌入和使用。

轻量级设计，快速生成模型图，适用于大规模模型和复杂网络。

专为研究人员设计，支持复杂模型结构可视化和学术成果展示。

只需几步，即可开始可视化你的PyTorch模型

git clone https://github.com/lyingloong/TorchNetViz.git
pip install -r requirements.txt

./scripts/powershell/Custom.ps1

使用 TorchNetViz 生成的模型结构图示例