sachinhosmani/torchvista

Interactive Pytorch forward pass visualization in notebooks

What it solves

它提供了一種在基於網頁的 notebook（如 Jupyter、Colab 與 Kaggle）內直接視覺化 PyTorch 模型前向傳播的方法，只需一行程式碼，即可更輕鬆地理解與除錯模型架構。

How it works

透過使用 trace_model 函式，工具會在給定特定輸入時追蹤模型的執行，並產生互動式圖形。此圖形允許使用者互動式探索模型結構，包括巢狀模組與參數資訊。

Who it’s for

PyTorch 開發者與研究人員，需在開發與除錯過程中視覺化模型的資料流與架構。

Highlights

  • Interactive Visualization：支援拖曳縮放與可折疊節點，以呈現階層式模組。
  • Error-Tolerant：即使發生形狀不匹配等錯誤，也能提供部分可視化，協助除錯。
  • Detailed Node Info：點擊節點即可顯示參數與屬性資訊。
  • Export Options：圖形可匯出為 PNG、SVG 或 HTML 檔案。
  • Compressed View：實驗性功能，將 Sequential 或 ModuleList 中相同類型的重複節點合併為單一區塊。