pytorch模型可视化：pytorchviz

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

1. 安装依赖和pytorchviz

   
   

     
     

      
      

     
     
     
     

      
      

       
       pip install graphviz
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       pip install torchviz （或pip install git+https://github.com/szagoruyko/pytorchviz）
      
      
     
     
   
   

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

   
   

     
     

      
      

     
     
     
     

      
      

       
       from torchviz 
       
       import make_dot
      
      
     
     

     
     

      
      

     
     
     
     

      
      
 
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       y = model(torch.rand(
       
       1,
       
       3,
       
       32,
       
       32).cuda())
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       g = make_dot(y,params=
       
       dict(model.named_parameters()))
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       g.view() 
       
       # 会在当前目录下保存一个“Digraph.gv.pdf”文件，并在默认浏览器中打开
      
      
     
     
   
   

遇到了以下问题 

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

sudo apt-get install graphviz

   
   

3. 拓展

   
   

     
     

      
      

     
     
     
     

      
      
 
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       # 以下三种方式都可以
      
      
     
     

     
     

      
      

     
     
     
     

      
      
 
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       g = make_dot(y)
      
      
     
     

     
     

      
      

     
     
     
     

      
      
 
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       # g=make_dot(y, params=dict(model.named_parameters()))
      
      
     
     

     
     

      
      

     
     
     
     

      
      
 
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       #g = make_dot(y, params=dict(list(model.named_parameters()) + [('x', x)]))
      
      
     
     

     
     

      
      

     
     
     
     

      
      
 
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       # 保存模型，以下两种方法都可以
      
      
     
     

     
     

      
      

     
     
     
     

      
      
 
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       # g.view() # 会生成一个 Digraph.gv.pdf 的PDF文件
      
      
     
     

     
     

      
      

     
     
     
     

      
      
 
      
      
     
     

     
     

      
      

     
     
     
     

      
      

       
       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-2_{aggregatepage}first_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-2_{aggregatepage}first_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