[P] TorchExplorer: the interactive neural network visualizer

​

https://preview.redd.it/dou086mdro6c1.png?width=3558&format=png&auto=webp&s=a59731b347fa68b61d45203e692846764ec2ee82

Hey all, I made a PyTorch [tool](https://github.com/spfrommer/torchexplorer) to interactively inspect activation/parameter histograms during training ([live demo here](https://api.wandb.ai/links/spfrom_team/8qqsxx9f)). It's sort of a combination of Netron (+ interactivity & nested submodules) and wandb.watch. It can:

* Interactively traverse model architectures, showing input/output tensor sizes and module parameters on hover
* Visualize module input/output tensors, parameters, and associated gradients as histograms over the course of training by drag-and-dropping nodes (modeled off of wandb.watch)
* Directly integrate with weights and biases or …

architectures biases call course histograms interactive machinelearning network neural network parameters serve simple tensor training wandb weights and biases