Combining Lux.jl with IPNewton() for Optimization

It’s one line, convert(CType,p) where CType is the typeof(ComponentArray(nn_p)) which can just be computed at the Lux.setup outside of the loss function so CType can just be a constant. It’s literally one line, delete all of this.

4 Likes