Problems with Dense for Linear regression #1137
Unanswered
RodrigoAVargasHdz
asked this question in
Q&A
Replies: 1 comment
-
Hi @RodrigoAVargasHdz, I need a little bit more information to help you here:
|
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hi,
I am trying to optimize linear regression and NN using FLAX. Both models were created with the Dense function from flax.linen.
However, even for a simple 1D linear regression, I haven't managed to converge to an optimal solution.
I created a repo that contains my example, https://github.com/RodrigoAVargasHdz/lr_flax.git
Could you please point me to what are my mistakes?
Beta Was this translation helpful? Give feedback.
All reactions