Is it possible to unregister a Parameter from an instance of a nn.Module? Let’s say I want to go through all Conv2d layers of a network and replace all weight parameters with my own custom nn.module? I can’t simply re-assign the weight attribute with my own module as I get:
TypeError: cannot assign 'CustomWeight' as parameter 'weight' (torch.nn.Parameter or None expected)

(I don’t want to define a custom Conv2d layer because I don’t want users  of my application to change code on their side)