How to rewrite a jax model with stacked parameters
No Comments
using Flux function stacked_model(params, x) y = x for p in params y = p(y) end return y end params = [Dense(10, 10, relu) for _ in 1:5] x = rand(10) y
This website uses cookies so that we can provide you with the best user experience possible. Cookie information is stored in your browser and performs functions such as recognising you when you return to our website and helping our team to understand which sections of the website you find most interesting and useful.