add build graph networkx

waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

MIT License

1.79k stars 266 forks source link

Open Andrechang opened 2 years ago

Andrechang commented 2 years ago

added a function to build hiddenlayer graph as networkx graph hope it helps