raghakot / keras-vis

Neural network visualization toolkit for keras
https://raghakot.github.io/keras-vis
MIT License
2.98k stars 664 forks source link

Investigate support for pytorch #51

Open raghakot opened 7 years ago

raghakot commented 7 years ago

All we need is iteration over layers, ability to take gradients with respect to some input in a graph, and abstractions for common keras backend calls.

It might be possible to make the lib support pytorch once these abstractions are in place.