Last active
November 17, 2021 07:44
-
-
Save lintangsutawika/5bffa8e5143353f65dbd2ce12332fe63 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# 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