Hi @ptrblck, I’d love to. But I am not sure that you can debug it with some code snippets because it includes some other modules(like the custom convolution). If possible, I can send you the whole project for debugging.
Another thing is that when I try to run on one single GPU(previously 4 GPU), I won’t have OOM error.