Just thought of a possible workaround although its fairly hacky and I’m hoping there is a cleaner way to do it.
It would (at least in theory), be possible to modify the Conv2d class in /torch/nn/modules/conv.py so that the forward function could accept “weight” as a parameter if supplied, else simply use self.weight
Not super jazzed about needing to modify the pytorch source unless its really necessary. So hopefully someone with more than my admittedly short (one day) experience with pytorch will have a better solution.