Custom Flux.jl Layer Not Updating (problem with Flux.trainable?)

Classic. The MOMENT I posted the question, I realized the issue.

The parameter a needs to be an array, even if it’s just a single element. That’s literally it.

I’ll leave this post in the hopes that others can learn from my mild embarrassment.

3 Likes