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

[tgrassi]

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.