pytoch可視化網絡工具-pytorchviz

tensorflow的可視化工具是tensorboard,但是tensorboard在顯示網絡結構圖的時候，顯示特別混亂． 之前看了caffe的可視化網絡圖，很清晰，最近發現pytorch的網絡圖可視化工具，對網絡圖的顯示也很清晰． 第一步：安裝： 一行命令安裝 conda install -c anaconda graphviz python-graphviz 第二步：可視化代碼（最好自己單獨寫

>>阅读原文<<