tensorflow / tensorflow/tensorboard
Graph Visualizer: turning off node collapsing
Open
@psybuzz is already working on this.
Since Jan 23, 2021.
plugin:graph
theme:usability
- Dominant language
- TypeScript
- Stars
- 7.2k
- Forks
- 1.7k
- Avg merge
- 4d 22h
- Merged PRs (30d)
- 1
Description
TensorBoard collapses Relu-Relu_6 nodes together into one node which turns simple neural net graphs into a jumple
Example pbtxt here
from resnet-8 from https://github.com/tensorflow/models
Current TensorBoard

Here's after I hack it to stop collapsing Relu nodes. Note that softmax layer is now at the top as expected, instead of middle of the graph as before

Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.