使用GRAPHVIZ+TORCHVIZ来可视化模型Pytorch模型结构
可视化pytorch模型结构,生成树状图,可能是很多朋友想做的事情,今天分享一种实现方法。
环境搭建
安装Python环境依赖
安装Python环境依赖:
pip install graphviz # 安装graphviz
pip install git+https://github.com/szagoruyko/pytorchviz # 通过git安装torchviz
安装系统graphviz 运行环境
下载地址:graphviz.gitlab.io/_pages/Download/Download_windows.html
使用方法
- 加载并运行待可视化模型