Skip to content

Instantly share code, notes, and snippets.

@lintangsutawika
Last active November 17, 2021 07:44
Show Gist options
  • Save lintangsutawika/5bffa8e5143353f65dbd2ce12332fe63 to your computer and use it in GitHub Desktop.
Save lintangsutawika/5bffa8e5143353f65dbd2ce12332fe63 to your computer and use it in GitHub Desktop.
# Run flax mnist example (optional)
pip install --user tensorflow-datasets==3.1.0 ml_collections clu
git clone https://github.com/google/flax.git
pip install --user -e flax
cd flax/examples/mnist
mkdir /tmp/mnist
python3 main.py --workdir=/tmp/mnist --config=configs/default.py --config.learning_rate=0.05 --config.num_epochs=5
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment