Dear AG1991,
Of course I would be pleasure to help but as I mentioned in “Custom nn.Conv2d”, backward pass had some complexity, so, I changed the solution based on what I expected from my code.
I mean that I did not customize convolution layer. I used conv2d function generated in Pytorch. If you want to customize convolution layer, please refer to the mentioned post(“Custom nn.Conv2d” ).
Good Luck 