We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
odeint
Hi! I've came across an example that runs on the GPU but doesn't run on the CPU. The cause seems to be x.norm() returning a nan when on CPU and a correct result on a GPU. (https://github.com/rtqichen/torchdiffeq/blob/master/torchdiffeq/_impl/misc.py#L74) This is happening when setting initial step at https://github.com/rtqichen/torchdiffeq/blob/master/torchdiffeq/_impl/misc.py#L126
x.norm()
nan
Don't have time to follow through on this at the moment, leaving it here for now.
The text was updated successfully, but these errors were encountered:
How did you set it to use GPU or CPU?
Sorry, something went wrong.
I guess the adjoint method doesn't support GPU. odeint_adjoint is referred as odeint in the mnist example if you set the adjoint parameter to be true.
No branches or pull requests
Hi!
I've came across an example that runs on the GPU but doesn't run on the CPU.
The cause seems to be
x.norm()
returning anan
when on CPU and a correct result on a GPU.(https://github.com/rtqichen/torchdiffeq/blob/master/torchdiffeq/_impl/misc.py#L74)
This is happening when setting initial step at
https://github.com/rtqichen/torchdiffeq/blob/master/torchdiffeq/_impl/misc.py#L126
Don't have time to follow through on this at the moment, leaving it here for now.
The text was updated successfully, but these errors were encountered: