Hi all! I am currently trying to create UDEs using a GPU backend for the neural network only. I have been following the examples here, but have not been able to get anything to work. I was wondering if anyone might know of or be willing to provide an alternative example of this, as I suspect this one might be out of date. Thank you!
19fitzna
1
That code from early 2020 would need some updates. GPU-based MNIST Neural ODE Classifier · DiffEqFlux.jl might be a better starting point?
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Struggling to train a UDE model with a GPU | 3 | 426 | February 14, 2023 | |
| MethodError: no method matching getrf! when I try to use GPU to training a universal differential equation with ForwardDiff.jl | 5 | 353 | December 6, 2022 | |
| Struggling to train a UDE with DiffEqGPU | 9 | 896 | March 7, 2023 | |
| Minimum Working Example (MWE) showing error in Universal Differential Equation (UDE) implementation | 10 | 489 | February 11, 2025 | |
| GPU simulation time of neural ODE program | 8 | 940 | August 16, 2019 |