I am trying to implement a CNN visualization technique 6. I essentially need a list of operations executed in the order of the forward pass, that the algorithm can work through backwards.
Due to the nature of the algorithm, operations such as pooling and reshaping must also be registered.
I could maybe have the user manually make a list of operations, but it isn’t ideal as I also use forward hooks to grab data between the modules.
I found a topic 11 that said I could maybe use the jit tracer, but I couldn’t see how that could be done from the documentation.
As I am trying to visualize information about the network, I don’t mind doing a trace for each run through the forward pass, as I would need to run the visualization algorithm each time anyways.
I understand there might not be a pretty way to do it, but I would appreciate any idea that could potentially work!