@albanD I have checked the code for indentation errors. In-fact I have just used the same class written for linear layers by replacing
F.linear in forward() to F.conv2d() and
making sure modifyW() returns the weights in correct shape
I will update here in case I could make it work with @property.
Best