aliaksandrsiarohin / cuda-gridsample-grad2 Goto Github PK

Hi @AliaksandrSiarohin,
thank you for this great project and your efforts on this! I was wondering under which license your code is available, since I could not find any information on this in the repo. Would you care to share any info on this and add a license to the repo?

Thanks for your work! It really works on my single GPU!!! However, I found it can't converage in multi-gpus. Have you tested this code on multi-gpus?

Hi, Siarohin. I'm using cuda_gridsample in my code to calculate the second backward of gridsample, but whenever i'm trying to save my torch model, an unexpected error always occur, i have tried torch.save() and dill.dump(), but neither of them ran successfully, do you have any idea what happened, any suggestions will be helpful.
Below are the error i encounted while running my code. I'm using python 3.9.13 and torch 1.10.0+cu113.