I have defined a MLP (multi layer perceptron) with the Chain command, and this MLP is wrapped in a mutable structure. The gradient of this Chain (2 layer dense NN) returns nothing. Is this a bug, or am I doing this the wrong way? Minimum working example: function loss(filter, x, y) return nor…

Change ps = Flux.Params(filt.f) to ps = Flux.params(filt.f)

Flux: gradient of Chain returns nothing

CarloLucibello June 11, 2021, 9:37am 2

Change

ps = Flux.Params(filt.f)

ps = Flux.params(filt.f)

2 Likes

Topic		Replies	Views
How to manually update the params in a `Flux.Chain` neural network? General Usage flux , machine-learning	1	986	July 19, 2020
Take gradient of parameters not working Machine Learning question , flux	0	354	January 12, 2021
Gradients of custom functions with Flux Machine Learning flux	2	1206	April 5, 2020
Flux.gradient returns dict of param and Nothing Machine Learning flux	3	786	September 22, 2021
Gradients for recurrent models in Flux.jl General Usage question	0	506	May 8, 2018