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
Train flux struct with list of models General Usage flux	8	423	March 31, 2023
Flux.functor New to Julia flux , functors	1	1529	April 10, 2020
Stacking layers example Flux - Flux.params empty New to Julia question	2	849	December 12, 2019
Problem on model and gradient descend in Flux General Usage	18	297	October 27, 2024
Possible anonymous function bug in Flux Machine Learning flux	9	720	April 14, 2020