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