sachinhosmani/torchvista

What it solves

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 내 동일 유형의 반복 노드를 하나의 블록으로 묶습니다.