Hi, I am doing a machine learning using gradient descend # I have a set of data, called samples # I have two models: model1 and model2 global par = Flux.params(model1, model2) #wrap them in params gs = gradient(par) do global l1 = mean(Flux.mse(model(sample),true1)) for sample in sa…

In your do block, the returned value is the value of the last expression. In this case, you actually return the output of a call to println, which is nothing (and it has no gradient). The last line of the do block needs to be return allLoss, and it should work?

Problems relating to "gradient is nothing"

General Usage

gdalle October 29, 2024, 2:14pm 11

See also

Did this solve your issue?

1 Like

Topic		Replies	Views
Flux.gradient returns dict of param and Nothing Machine Learning flux	3	786	September 22, 2021
Params not getting updated during training New to Julia flux	25	1789	October 11, 2020
Computing Flux.gradient change the model Machine Learning	10	1395	January 31, 2021
Gradients of custom functions with Flux Machine Learning flux	2	1206	April 5, 2020
Take gradient of parameters not working Machine Learning question , flux	0	354	January 12, 2021

Problems relating to "gradient is nothing"

Related topics