pytorch模型可视化:pytorchviz

这个包似乎是用来做可视化的,用于辅助pytorch,多用pip以及conda换着装,名字也可以换成pytorchviz或者torchviz,或者换源,卸载重装等,多试几次,观察规律。。

1. 安装依赖和pytorchviz


   
     
     
     
     
  1. pip install graphviz
  2. pip install torchviz (或pip install git+https://github.com/szagoruyko/pytorchviz)

2. 使用如下代码,保存在当前目录下的pdf文件中,并打开


   
     
     
     
     
  1. from torchviz import make_dot
  2. y = model(torch.rand( 1, 3, 32, 32).cuda())
  3. g = make_dot(y,params= dict(model.named_parameters()))
  4. g.view() # 会在当前目录下保存一个“Digraph.gv.pdf”文件,并在默认浏览器中打开

遇到了以下问题 pytorch模型可视化:pytorchviz_第1张图片

原因是:未在系统中配置graphviz工具的环境变量,Graphviz是AT&T Labs Research开发的图形绘制工具软件,不是python 工具,因此,需要独立的在系统内安装graphviz,仅在python环境内安装组件是无法使用的的。

sudo apt-get install graphviz

   
     
     
     
     

3. 拓展


   
     
     
     
     
  1. # 以下三种方式都可以
  2. g = make_dot(y)
  3. # g=make_dot(y, params=dict(model.named_parameters()))
  4. #g = make_dot(y, params=dict(list(model.named_parameters()) + [('x', x)]))
  5. # 保存模型,以下两种方法都可以
  6. # g.view() # 会生成一个 Digraph.gv.pdf 的PDF文件
  7. g.render( 'espnet_model', view= False) # 会自动保存为一个 espnet.pdf,第二个参数为True,则会自动打开该PDF文件,为False则不打开

 

 

 

 

 

更多参考

https://blog.csdn.net/srf_code/article/details/103108862

https://download.csdn.net/download/weixin_42114041/18351051?utm_medium=distribute.pc_aggpage_search_result.none-task-download-2aggregatepagefirst_rank_ecpm_v1~rank_v31_ecpm-3-18351051.pc_agg_new_rank&utm_term=pytorchviz%E5%AE%89%E8%A3%85&spm=1000.2123.3001.4430

https://gitcode.net/mirrors/szagoruyko/pytorchviz?utm_medium=distribute.pc_aggpage_search_result.none-task-code_china-2aggregatepagefirst_rank_ecpm_v1~rank_v31_ecpm-4-6370.pc_agg_new_rank&utm_term=pytorchviz%E5%AE%89%E8%A3%85&spm=1000.2123.3001.4430&from_codechina=yes

你可能感兴趣的:(人工智能-神经网络,机器学习入门必备,Python使用技巧,人工智能)