Here’s how to track activation maxima without affecting weights:
✅ Use register_buffer for persistent storage
✅ Use register_forward_hook to capture activations
Here’s how to track activation maxima without affecting weights:
✅ Use register_buffer for persistent storage
✅ Use register_forward_hook to capture activations