I am training a NN with some synthetic data. I was running everything on REPL and things were going on the right direction, but when I tried to wrap my code inside a function() to use custom training loops, my model(x) started to return the same output every iteartion, as it would have stopped up…

Probably has to do with your loss(x, y) definition. I suspect the m in the function body is not referencing the m created inside all_the_code and instead referencing some other m in global scope. In any case, it is better to explicitly pass in the model to the loss function, because depending on the…

Training gets differents results when using Flux.train() inside function

Specific Domains Machine Learning

german_brunini August 23, 2020, 11:01pm 3

@darsnack, You nailed it. Thank you very much. Now it run like it’s supposed to!.

Topic		Replies	Views
Flux training GPU vs CPU different results Machine Learning gpu , flux	7	1192	August 20, 2020
Params not getting updated during training New to Julia flux	25	1789	October 11, 2020
Flux networks as arguments in functions Machine Learning flux	6	636	May 8, 2019
Flux.jl is not modifying the loss Machine Learning	4	815	June 25, 2019
How to use Flux.train! to train custom layer? Machine Learning question	2	1836	September 2, 2019

Training gets differents results when using Flux.train() inside function

Related topics