Yeah, that won’t work.
Could you share the code for CustomWeight?
If that’s not possible, could you explain, how you wanted to assign this module to the conv parameters?
I assume CustomWeight is not just storing the parameters, but also applies some operations. Is that correct?