How to display branch nodes / parallel blocks

waleedka / hiddenlayer

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

MIT License

1.79k stars 266 forks source link

How to display branch nodes / parallel blocks #86

Closed talhaanwarch closed 3 years ago

talhaanwarch commented 3 years ago

I am confused about how to transform the following structure which has branches coming out from a node. This is what I tried hl.transforms.Fold("""Conv > LeakyRelu > Conv > Concat > LeakyRelu """, "Fire","FireBlock")

HaoLuo2627 commented 3 years ago

I find this in demos: And the expressions Conv | MaxPool will match any Conv and MaxPool layers that are in parallel branches (i.e. have the same parent node). So I think """Conv > LeakyRelu > (Conv | Conv) > Concat > LeakyRelu """ may work.

talhaanwarch commented 3 years ago

Thanks. It worked.