I am trying to visualize my lstm network using torchviz, but the graph I get is very complicated due to the nature of lstm cells. Is there a way to make torchviz simplify it?
I have also tried using tensorboard, but I couldn’t get it to work, I think it’s because I am packing and unpacking my sequences.
