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

 
 
 

Contributors

Languages