JAX-Flax-Tutorial-Image-Classification-with-Linen

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

Search similar apps

    License

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

    Creator

    8bitmp3

    Related apps