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
Gradient not being computed when training NN using Flux Machine Learning question , flux , sciml	3	148	January 30, 2025
Flux.gradient returns dict of param and Nothing Machine Learning flux	3	810	September 22, 2021
Take gradient of parameters not working Machine Learning question , flux	0	359	January 12, 2021
How to manually update the params in a `Flux.Chain` neural network? General Usage flux , machine-learning	1	991	July 19, 2020
Gradients of custom functions with Flux Machine Learning flux	2	1220	April 5, 2020