SPN: cudaCheckError() failed : an illegal memory access was encountered

[youmi-zym]

Hi,
Thanks for sharing your code, when I use spn it often throws an error: cudaCheckError() failed: an illegal memory access was encountered.
Have you ever encountered?

Running env:

[mileyan]

Hi, are you using pytorch 1.0?

[youmi-zym]

Hi, I'm using pytorch1.2. The problem I met is: if just train or test your model, it goes ok. However, if I iterate the model for more than 1 time, it gives an error: cudaCheckError() failed: illegal memory access was encountered. I guess the error happens since the memory doesn't release after a full iteration, and when iterating the second time, it accesses the memory of the first iteration.

[mileyan]

I think you are right, I guess pytorch might change some of its C extension APIs in version 1.2. Now, this code only supports pytorch 0.4 or 1.0. I will work on it this summer to support pytorch 1.4.