如何可视化PyTorch的网络模型计算图和后传路径图（可视化PyTorch网络结构）
PyTorchViz包（https://github.com/szagoruyko/pytorchviz）可以用来方便地绘制PyTorch正向网络计算图和后传计算路径，本文讲解如何使用PyTorchViz包，主要参考网站包括：
- https://towardsdatascience.com/understanding-pytorch-with-an-example-a-step-by-step-tutorial-81fc5f8c4e8e
- https://stackoverflow.com/questions/52468956/how-do-i-visualize-a-net-in-pytorch
1. 安装PyTorchViz
pip install torchviz
安装中间如果出现任何问题，请自行google或者留言，我安装过程中出现的错误已经忘记是什么了……不过问题不大。
2. 导入绘制函数
from torchviz import make_dot
3. 定义网络并前向计算输入
4. 绘制计算图和后传路径
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！