Skip to content

Commit

Permalink
Rename linen_examples -> examples part 2:
Browse files Browse the repository at this point in the history
Add changes that weren't accounted for in d068512
  • Loading branch information
avital committed Jan 19, 2021
1 parent d068512 commit 4d593fe
Show file tree
Hide file tree
Showing 6 changed files with 12 additions and 33 deletions.
17 changes: 0 additions & 17 deletions .github/workflows/apply-howto-branches.yml

This file was deleted.

3 changes: 0 additions & 3 deletions .github/workflows/build.yml
Expand Up @@ -37,9 +37,6 @@ jobs:
- name: Test with pytest and generate coverage report
run: |
tests/run_all_tests.sh --with-cov
- name: Check HOWTO patches
run: |
howtos/scripts/check_howto_diffs.sh
- name: Upload coverage to Codecov
uses: codecov/codecov-action@v1
with:
Expand Down
20 changes: 10 additions & 10 deletions docs/examples.rst
Expand Up @@ -7,8 +7,8 @@ Core examples
Each example is designed to be self-contained and easily forkable, while
reproducing relevant results in different areas of machine learning. These
examples adhere to a shared style and functionality outlined in `#231`_. All
examples are under the folder `flax/linen_examples/
<https://github.com/google/flax/tree/master/linen_examples/>`__. Some of the
examples are under the folder `flax/examples/
<https://github.com/google/flax/tree/master/examples/>`__. Some of the
examples below have a link [Interactive] that lets you run them directly in
Colab.

Expand All @@ -17,35 +17,35 @@ Colab.

Image classification

- `MNIST <https://github.com/google/flax/tree/master/linen_examples/mnist/>`__ [`Interactive
<https://colab.research.google.com/github/google/flax/blob/master/linen_examples/mnist/mnist.ipynb>`__] :
- `MNIST <https://github.com/google/flax/tree/master/examples/mnist/>`__ [`Interactive
<https://colab.research.google.com/github/google/flax/blob/master/examples/mnist/mnist.ipynb>`__] :
Convolutional neural network for MNIST classification (featuring simple code).
- `ImageNet <https://github.com/google/flax/tree/master/linen_examples/imagenet/>`__ :
- `ImageNet <https://github.com/google/flax/tree/master/examples/imagenet/>`__ :
Resnet-50 on imagenet with weight decay (featuring multi host SPMD, custom
preprocessing, checkpointing, dynamic scaling, mixed precision).

Reinforcement Learning

- `Proximal Policy
Optimization <https://github.com/google/flax/tree/master/linen_examples/ppo/>`__ :
Optimization <https://github.com/google/flax/tree/master/examples/ppo/>`__ :
Learning to play Atari games (featuring single host SPMD, RL setup).

Natural language processing

- `Sequence to sequence for number
addition <https://github.com/google/flax/tree/master/linen_examples/seq2seq/>`__
addition <https://github.com/google/flax/tree/master/examples/seq2seq/>`__
(featuring simple code, LSTM state handling, on the fly data generation).
- `Transformer model on
WMT <https://github.com/google/flax/tree/master/linen_examples/wmt/>`__ :
WMT <https://github.com/google/flax/tree/master/examples/wmt/>`__ :
Translating English/German (featuring multihost SPMD, dynamic bucketing, attention cache,
packed sequences, recipe for TPU training on GCP).

Generative models

- `Variational
auto-encoder <https://github.com/google/flax/tree/master/linen_examples/vae/>`__ :
auto-encoder <https://github.com/google/flax/tree/master/examples/vae/>`__ :
Trained on binarized MNIST (featuring simple code, vmap).
- `PixelCNN++ <https://github.com/google/flax/tree/master/linen_examples/pixelcnn/>`__ :
- `PixelCNN++ <https://github.com/google/flax/tree/master/examples/pixelcnn/>`__ :
Trained on cifar10 (featuring single host SPMD, checkpointing, Polyak decay).


Expand Down
4 changes: 2 additions & 2 deletions docs/notebooks/linen_intro.ipynb
Expand Up @@ -38,9 +38,9 @@
"\n",
"⟶ [Slides](https://docs.google.com/presentation/d/1ngKWUwsSqAwPRvATG8sAxMzu9ujv4N__cKsUofdNno0/edit?usp=sharing) for the core ideas of the new Functional Core and Linen\n",
"\n",
"\"Design tests\" guided our design process. Many are available for [functional core](https://github.com/google/flax/tree/linen2/linen_examples/core_design_test) and some for the [proposed Module abstraction](https://github.com/google/flax/tree/linen2/linen_examples/linen_design_test/)\n",
"\"Design tests\" guided our design process. Many are available for [functional core](https://github.com/google/flax/tree/master/examples/core_design_test) and some for the [proposed Module abstraction](https://github.com/google/flax/tree/master/examples/linen_design_test/)\n",
"\n",
"⟶ Ported examples: [ImageNet](https://github.com/google/flax/tree/linen2/linen_examples/imagenet) and [WMT](https://github.com/google/flax/tree/linen2/linen_examples/wmt) (to the proposed Module abstraction). TODO: Port to functional core.\n",
"⟶ Ported examples: [ImageNet](https://github.com/google/flax/tree/master/examples/imagenet) and [WMT](https://github.com/google/flax/tree/master/examples/wmt) (to the proposed Module abstraction). TODO: Port to functional core.\n",
"\n",
"⟶ Our new [discussion forums](https://github.com/google/flax/discussions/)\n",
"\n"
Expand Down
File renamed without changes.
1 change: 0 additions & 1 deletion linen_examples/vae/results/.gitignore

This file was deleted.

0 comments on commit 4d593fe

Please sign in to comment.