NeuralPDE.jl GPU example returns "ERROR: UndefVarError: `Adam` not defined"

Hello everyone,
I’m trying to run the example from NeuralPDE:
https://neuralpde.sciml.ai/stable/tutorials/gpu/

However when I run the line:

res = Optimization.solve(prob,Adam(0.01);callback = callback,maxiters=2500)

I get the error:

ERROR: UndefVarError: `Adam` not defined
Stacktrace:
 [1] top-level scope
   @ Untitled-2:71

Not sure how to proceed.

I suspect you skipped the line.

using OptimzationOptimisers
1 Like