Skip to content

amacbride/pyviz

Repository files navigation

pyviz

Quick PyTorch visualizer app

Features:

  • Interactive 3D visualization with orbit controls
  • Color-coded neurons by layer type (green=input, blue=hidden, red=output)
  • Weighted connections between layers
  • Network architecture info panel
  • Randomize activations button
  • Toggle connections visibility

Usage: source pyviz/bin/activate python main.py # Default network python main.py --custom # Custom network python main.py --port 9000 # Custom port

The app starts a local web server and opens the 3D visualization in your browser.

Simple demo

About

Quick PyTorch visualizer app

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages