I have a model and would like to regularize the activations of one of its layers. I wondered if I had to do something “special” to get them or just returning the layer I need from forward() would work?
I would appreciate if anyone could give me a hint about it. I probably am overthinking this.