-
Hello,
How to train the autoregressive model for inference? In the forward function, what would be the first expression_one_hot tensor? I understand subsequent inputs would be the labels output of …
-
This issue tracks updates the `pixelcnn` example to follow practices outlined in #231.
- [x] Port to linen API - once ported all subsequent changes should be done in `linen_examples/pixelcnn`
- […
-
Following some discussion I wrote down a set of guidelines for Flax examples inside the google/flax repository. It will take some work to update the current set of examples but I believe that the exam…
-
The PixelCNN has an example of using custom gradients: [pixelcnn.py](https://github.com/google/flax/blob/master/examples/pixelcnn/pixelcnn.py#L422).
[The Autodiff Cookbook](https://jax.readthedocs.…
-
See https://github.com/google/flax/blob/master/docs/flip/1009-optimizer-api.md#update-plan
The following examples need to be updated
- [x] imagenet #1251
- [x] mnist #1302
- [x] nlp_seq #191…
-
Some of our examples have additional dependencies (tensorflow_text for WMT, pillow for PixelCNN). Currently we define these in 3 locations:
- [tests_requires in the setup.py](github.com/google/flax/b…
-
Maybe I am not looking hard enough but I can't seem to find the code for density estimation using pixelcnn++. Can you direct me to the file for the same?
-
This task is to try to implement a generative model for 1D data
Here are a few pointers:
- [notebook](https://colab.research.google.com/drive/1wPNS0gBetVUk2EAK8_5HJmrAT_o476L0?usp=sharingv) with…
EiffL updated
2 years ago
-
In some of our examples we `jax.jit` the `init()`, for instance in the [WMT example](https://github.com/google/flax/blob/master/linen_examples/wmt/train.py#L472), and in other examples we don't.
@M…
-
Dear devs,
The example at https://keras.io/examples/generative/vq_vae/ works only with tfp up until v0.13.0. https://github.com/keras-team/keras-io/pull/911 saw a resolution of this issue, but the …