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
→

Metadata

How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).

Results 2 JAX-Flax-Tutorial-Image-Classification-with-Linen issues
Sort by recently updated
recently updated
newest added

Error while running CNN Python notebook with CIFAR10 Images

Colab_The_annotated_MNIST_image_classification_example_with_Flax_Linen_and_Optax.ipynb In the above notebook, I ran the same code as it is with CIFAR10 image dataset from tfds. There was no problem with the dataset itself, but in the...

AbhiDu96 avatar AbhiDu96

Trainning problems

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...

Khazan1 avatar Khazan1

← Metadata

18
Stars
2
Forks
Watchers

Owner

verified publisher icon 8bitmp3

Metadata

How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).

Back