【python技巧】pytorch网络可视化

参考

https://blog.csdn.net/qq_40726937/article/details/106122082

1. graphviz + torchviz

环境安装简单

pip install torchviz
pip install graphviz

代码

import torch
from torchvision import model
from torchviz import make_dot

models = models.resnet18()
x = torch.randn(1,3,224,224)
net_plot = make_dot(model(x),params = dict(models.named_parameters()))
net_plot.view()

缺点:导出的是pdf

相关推荐

  1. python技巧pytorch网络

    2024-07-09 17:50:10       7 阅读
  2. Python

    2024-07-09 17:50:10       16 阅读
  3. Python 数据

    2024-07-09 17:50:10       34 阅读

最近更新

  1. PHP MySQL 读取数据

    2024-07-09 17:50:10       0 阅读
  2. Handler续谈(epoll)

    2024-07-09 17:50:10       0 阅读
  3. Git提交到错误分支怎么办?(解决办法)

    2024-07-09 17:50:10       0 阅读
  4. 在ubuntu系统上安装qt 2

    2024-07-09 17:50:10       0 阅读
  5. CONFIG_MTD_SPI_NOR_USE_4K_SECTORS

    2024-07-09 17:50:10       0 阅读
  6. 网络通信协议

    2024-07-09 17:50:10       0 阅读
  7. opencv 使用XML和YAML格式来输入输出文件

    2024-07-09 17:50:10       0 阅读

热门阅读

  1. 单例模式的实现

    2024-07-09 17:50:10       5 阅读
  2. 【MIT 6.5840/6.824】Lab1 MapReduce

    2024-07-09 17:50:10       6 阅读
  3. 【云原生】Kubernetes之持久化

    2024-07-09 17:50:10       4 阅读
  4. urlib Python爬虫

    2024-07-09 17:50:10       6 阅读
  5. 【MySQL】SQL中的DROP、DELETE和TRUNCATE的区别

    2024-07-09 17:50:10       9 阅读
  6. 云原生监控-Kubernetes-Promethues-Grafana

    2024-07-09 17:50:10       8 阅读
  7. arm (exti中断)

    2024-07-09 17:50:10       9 阅读
  8. LRU Cache 双向链表以及STL list实现----面试常考

    2024-07-09 17:50:10       12 阅读