Pytorch 模型结构图可视化

最新推荐文章于 2025-04-16 17:32:05 发布

曼车斯基

最新推荐文章于 2025-04-16 17:32:05 发布

阅读量610

点赞数

分类专栏： Pytorch 文章标签： graphviz

本文链接：https://blog.youkuaiyun.com/redredxcar/article/details/115300946

版权

Pytorch 专栏收录该内容

4 篇文章

订阅专栏

本文档介绍了如何在Mac上安装Graphviz，解决安装过程中的问题，以及如何利用PyTorch库torchviz，将一个多层感知器（MLP）模型的计算图进行可视化展示。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 安装graphviz

brew install graphviz

我安装过程中碰到了这个错误：

Directory not empty @ dir_s_rmdir - /usr/local/opt/openssl

需要卸载再重装 openssl：

sudo rm -rf /usr/local/opt/openssl

brew install openssl

2. pip install graphviz

3. 使用graphviz 可视化一个mlp

import torch
from torch import nn
from torchviz import make_dot
from torch.autograd import Variable

model = nn.Sequential()
model.add_module('W0', nn.Linear(8, 16))
model.add_module('tanh', nn.Tanh())
model.add_module('W1', nn.Linear(16, 1))

x = Variable(torch.randn(1,8))

vis_graph = make_dot(model(x), params=dict(model.named_parameters()))
vis_graph.view()

结果：

模型结构图