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	794	September 22, 2021
Params not getting updated during training New to Julia flux	25	1800	October 11, 2020
Computing Flux.gradient change the model Machine Learning	10	1396	January 31, 2021
Gradients of custom functions with Flux Machine Learning flux	2	1209	April 5, 2020
Take gradient of parameters not working Machine Learning question , flux	0	356	January 12, 2021

Problems relating to "gradient is nothing"

Related topics