When displaying a graph using torch.utils.tensorboard.SummaryWriter.add_graph
, only modules end up in boxes. I use a lot of ops around it, and it can be really, really hard to tell what’s going on. Is there something like tensorflow1.0’s tf.namespace
(might be variable_scope
actually), which would let me sort operations to more clearly see what’s happening? Thanks.
@Nathan_Wood Have you solved this issue ?
I have similar issue at Adding node and cell names for tensorboard graph