I am currently implementing a toy version of the architecture as found in: Deep learning for universal linear embeddings of nonlinear dynamics | Nature Communications , which is effectively an encoder → affine → decoder process. I have already implemented the architecture and have been testing on …

I was referring to the MWE, in which only model.K is used. Incidentally, the gradient there is nothing as well in the MWE because of an interaction between implicit parameters (i.e. the thing you get from params and pass to gradient) and the AD (Zygote). c.f: julia> using Zygote, LinearAlgebra jul…

Flux.jl Inconsistent Training on Custom Architecture

New to Julia

than123 June 22, 2021, 1:47am 3

I see! Will do. Thanks for letting me know.

Topic		Replies	Views
Incorrect (implicit) gradients for a custom model in Flux Machine Learning question , flux	3	450	March 12, 2021
Another error with defining Flux.jl gradients Machine Learning first-steps , flux	5	1682	December 5, 2018
Gradients of custom functions with Flux Machine Learning flux	2	1206	April 5, 2020
Params not getting updated during training New to Julia flux	25	1789	October 11, 2020
Error with defining customer gradients in Flux.jl Machine Learning flux	9	1722	November 21, 2018

Flux.jl Inconsistent Training on Custom Architecture

Related topics