Flux.params does not recognize parameters with `x -> layer(x)` syntax

Thanks @tomerarnon, it works!
(updated Julia to version 1.5.1 to make your code working).

I will link your answer to the other issue.

1 Like