pytorch網絡結構模型可視化—學習筆記

• 安裝graphviz

pip3 install graphviz --user

• 安裝pytorchviz

pip3 install git+https://github.com/szagoruyko/pytorchviz --user

• 測試

from torchviz import make_dot

• 使用

from torchviz import make_dot
...
atts, dets = model(images)
vis_graph = make_dot(model(images), params=dict(model.named_parameters()))
vis_graph.view()

• 結果
  
• 參考資料
  使用pytorchviz進行pytorch執行過程的可視化