Hey,
I’m wondering if it is possible in Tensorboard with pytorch to define groups of nodes that are displayed as one node. E.g. in TensorFlow it is possible to group and name nodes in Tensorboard using tf.name_scope (as far as I understood) as explained here 1. This is very helpful for having a better overview if you have several ‘units’ in your graph. A simple example, in a feed forward NN I could wish to group all the hidden layers in a node called ‘representation’ and the output layer in a node called ‘classifier’.
Is this somehow possible in pytorch? For me it seems like a very important feature yet I could’t find anything.
Cheers,
Spyderman