Is it possible to modify module weights with a backward hook or would it mess up the gradients? If not, is there a way to modify a specific module’s weights without having to search through the entire network using apply? I’m thinking of something like clipping weights with a backward hook.