Changing cuda to True seems to cause the code to stop working correctly

[rbunn80110]

This project is really great work! However, I seem to be struggling trying to get it to work using cuda. First if you change cuda to True, then there is an error related to needing to transfer the prediction variable to the cpu. Meaning you have to add .cpu() to the prediction variable to get it to at least run. However, at this point things appear to never move towards converging. Even after running hundreds of iterations. What all has to be changed to execute the examples with cuda and have them converge? Running the code on the CPU works beautifully though.

[staturecrane]

Hi @rbunn80110. Could you tell me the example you are trying to use with CUDA?

[rbunn80110]

The lander example