pytorch 模型可视化的例子
2019-08-17 17:39
1426 查看
如下所示:
一. visualize.py
from graphviz import Digraph import torch from torch.autograd import Variable def make_dot(var, params=None): """ Produces Graphviz representation of PyTorch autograd graph Blue nodes are the Variables that require grad, orange are Tensors saved for backward in torch.autograd.Function Args: var: output Variable params: dict of (name, Variable) to add names to node that require grad (TODO: make optional) """ if params is not None: assert isinstance(params.values()[0], Variable) param_map = {id(v): k for k, v in params.items()} node_attr = dict(style='filled', shape='box', align='left', fontsize='12', ranksep='0.1', height='0.2') dot = Digraph(node_attr=node_attr, graph_attr=dict(size="12,12")) seen = set() def size_to_str(size): return '('+(', ').join(['%d' % v for v in size])+')' def add_nodes(var): if var not in seen: if torch.is_tensor(var): dot.node(str(id(var)), size_to_str(var.size()), fillcolor='orange') elif hasattr(var, 'variable'): u = var.variable name = param_map[id(u)] if params is not None else '' node_name = '%s\n %s' % (name, size_to_str(u.size())) dot.node(str(id(var)), node_name, fillcolor='lightblue') else: dot.node(str(id(var)), str(type(var).__name__)) seen.add(var) if hasattr(var, 'next_functions'): for u in var.next_functions: if u[0] is not None: dot.edge(str(id(u[0])), str(id(var))) add_nodes(u[0]) if hasattr(var, 'saved_tensors'): for t in var.saved_tensors: dot.edge(str(id(t)), str(id(var))) add_nodes(t) add_nodes(var.grad_fn) return dot
二. 使用步骤
import torch from torch.autograd import Variable from models import * from visualize import make_dot x = Variable(torch.rand(1, 3, 256, 256)) model = GeneratorUNet() y = model(x) g = make_dot(y) g.view()
三. 效果展示
以上这篇pytorch 模型可视化的例子就是小编分享给大家的全部内容了,希望能给大家一个参考
您可能感兴趣的文章:
相关文章推荐
- pytorch借助tensorboard实现模型可视化
- 模型训练过程初步可视化
- 【caffe源码研究】第二章:使用篇(5) : 模型可视化
- PLUTO平台是由美林数据技术股份有限公司下属西安交大美林数据挖掘研究中心自主研发的一款基于云计算技术架构的数据挖掘产品,产品设计严格遵循国际数据挖掘标准CRISP-DM(跨行业数据挖掘过程标准),具备完备的数据准备、模型构建、模型评估、模型管理、海量数据处理和高纬数据可视化分析能力。
- caffe可视化模型
- caffe模型可视化
- 从一个工作流流转中人员设定的例子看WBISF流程引擎所定义的用户授权安全模型
- 惊艳全球数据行业的16个数据可视化例子
- keras可视化模型结构
- PyTorch加载预训练模型的问题
- pytorch的一些例子
- css 样式设计(一)( 在线150个例子 | 背景 | 文本 | 字体 | 链接 | 列表 | 表格 | 盒模型 | 边框 | 轮廓 | 边距 | 填充 |分组和嵌套 | 尺寸 | 定位 | 浮动 |对齐 )
- 说说标准——CSS核心可视化格式模型(visual formatting model)之十三:分层的显示(Layered presentation)
- 变分推断学习笔记(2)——一维高斯模型的例子
- 生产者消费者模型的四种java编程例子
- epoll模型与select模型的区别(宿管大妈的例子)
- ROS学习笔记------ROS基础-----基础编程 day 5 2019/3/1 帅某(QT工具箱,rviz可视化平台,gazebo本地模型保存)
- NetScope:基于 prototxt 可视化模型结构
- Caffe模型可视化
- tensorflow和pytorch模型之间转换