Hi weedwind,
I didn’t see any masking work in your description. I am doing like this : Padding and masking in convolution 450
Currently, my model seems working well but I have some questions about gradient too.