Flux’s typical go to when writing models is Chains, and for most feedforward models (with the occasional skip connection), with one input, it works quite beautifully. Problem is, when defining models as functions, we lose the ability to get their params directly, or move them to the GPU. Even as th…

Writing complex Flux Models

dunefox June 5, 2020, 7:14pm 5

Doesn’t putting the model struct into a module solve that problem? I think you can redefine it then.

Topic		Replies	Views
Flux.functor New to Julia flux , functors	1	1519	April 10, 2020
Stacking layers example Flux - Flux.params empty New to Julia question	2	846	December 12, 2019
Train flux struct with list of models General Usage flux	8	399	March 31, 2023
Possible anonymous function bug in Flux Machine Learning flux	9	703	April 14, 2020
Flux; `params` cannot track any parameters of networks when building networks without `Chain` General Usage flux	6	635	December 31, 2020