Hi,thanks for ur work.
I got a issue when I tried to apply ur method to my model:
RuntimeError: Function CudnnConvolutionBackward returned an invalid gradient at index 1 - got [96, 32, 3, 3] but expected shape compatible with [192, 32, 3, 3]
I already change the optimization to IsO, can u help to figure it out please?
Hi,thanks for ur work. I got a issue when I tried to apply ur method to my model: RuntimeError: Function CudnnConvolutionBackward returned an invalid gradient at index 1 - got [96, 32, 3, 3] but expected shape compatible with [192, 32, 3, 3] I already change the optimization to IsO, can u help to figure it out please?