willyfh/visualtorch

무엇을 해결하는가

VisualTorch는 PyTorch 신경망 아키텍처를 시각화하는 방법을 제공하여, Sequential 및 Custom 모델 모두의 구조를 더 쉽게 이해하고 문서화할 수 있게 합니다. 이는 수동으로 또는 단순한 텍스트 로그를 통해 시각화하기 어려운 복잡한 모델 아키텍처 문제를 해결합니다.

작동 방식

이 도구는 트레이싱 기반 접근 방식을 사용하여, 더미 입력을 통한 실제 포워드 패스를 수행하여 모델의 연결과 레이어를 매핑합니다. flow-style, graph-style, LeNet-style의 세 가지 뚜렷한 스타일로 다이어그램을 생성할 수 있습니다. 또한, 모델 아키텍처를 레이어별로 보여주는 애니메이션 GIF 공개 기능을 지원합니다.

대상 사용자

연구 논문, 문서화 또는 디버깅을 위해 신경망 아키텍처의 명확하고 전문적인 다이어그램이 필요한 PyTorch 사용자 연구자 및 개발자.

주요 특징

• 다양한 시각화 스타일: flow, graph, LeNet-style 다이어그램을 지원합니다.
• 애니메이션 공개: 모델 아키텍처를 애니메이션 GIF로 렌더링하는 기능입니다.
• 통합 트레이싱 백엔드: PyTorch 모델을 처리하기 위한 자체 트레이싱 로직을 특징으로 합니다.
• MCP Integration: stdio를 통해 모델 소스로부터 다이어그램을 생성하기 위한 선택적 MCP 서버를 포함합니다.
• 연구 입증됨: Nature, IEEE, MDPI 전반의 출판물에서 사용되었습니다.