Currently this is how i add a histogram for my layer’s weights:
How can I add the activation histogram in a similar manner?
My assumption would be that I have to add it in the forward function after it passes through the relevant ReLU, softmax, etc. However what do I do when i don’t have access to the forward function directly (model imported from torchvision.models, or defined as a
Example of what i’m looking for taken from tensorboard.dev: