willyfh/visualtorch

VisualTorch aims to help visualize Torch-based neural network architectures.

What it solves

VisualTorch는 PyTorch 신경망 아키텍처를 시각화하는 방법을 제공하여, 개발자와 연구자가 모델 구조를 이해하고, 문서화하고, 발표할 수 있도록 돕습니다. 단순한 순차적 모델부터 더 복잡한 커스텀 모델까지 모두 지원합니다.

How it works

이 도구는 트레이싱 기반(tracing-based) 방식을 사용하여, 더미 입력(dummy input)으로 실제 포워드 패스(forward pass)를를 수행하여 모델의 아키텍처를 매핑합니다. 그런 다음, 이 아키텍처를 플로우 스타일, 그래프 스타일, LeNet 스타일의 세 가지 뚜렷렷한 시각적 스타일로 렌더링할 수 있습니다. 또한, 모델이 레이어별로 구축되는 모습을 보여주는 애니메이션 GIF 공개 효과를를 생성할 수 있습니다.

Who it’s for

연구자 및 개발자를 포함하여, 논문 발표, 프레젠테이션 또는 내부 문서를 위해 신경망의 전문적인 다이어그램을 생성해야 하는 PyTorch 사용자를 위해 설계되었습니다.

Highlights

  • Multiple Visualization Styles: 플로우, 그래프, LeNet 스타일의 다이어그램을 지원합니다.

  • Animated Reveals: 모델 아키텍처를 애니메이션 GIF로 렌더링할 수 있는 기능이 있습니다.

  • MCP Integration: stdio를 통해 모델 소스 코드를 통해 다이어그램을การating 생성할 수 있는 선택 사항인 MCP server를 포함합니다.

  • Research-Backed: Nature, IEEE, 및 MDPI에 발표된 연구에서 사용되었습니다.