JAX-Flax-Tutorial-Image-Classification-with-Linen icon indicating copy to clipboard operation
JAX-Flax-Tutorial-Image-Classification-with-Linen copied to clipboard

Published 20 hours ago • verified publisher icon 8bitmp3

Trainning problems

Open Khazan1 opened this issue 2 years ago • 0 comments

Hello, I noticed that your code is the same as the getting start in the official jax document, but when I changed the more complex network and data and made a regression problem, I finally found that after the first batch entered fuc train_ step, no results are returned, and the training is not started, then the program is completed. Considering that you may be experienced in jax training, can you give me some advice or guidance on the above issues.Thanks for any apply.

Khazan1 avatar Dec 07 '22 08:12 Khazan1