Function CudnnConvolutionBackward returned an invalid gradient at index 1 - got [294, 592, 1, 1] but expected shape compatible with [294, 768, 1, 1]

Question

I\'m working on a pytorch (torch=1.2.0) project considering CNN\'s pruning. I dropped few filters from a model and tried to retrain the reduced model. However my code crashe