Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
didriknielsen authored Sep 22, 2020
1 parent 236c3dd commit 5d81a8e
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,7 @@ model = Flow(base_dist=StandardNormal((2,)),
AffineCouplingBijection(net()), ActNormBijection(2),
])
```
See [here](https://github.com/didriknielsen/survae_flows/blob/master/examples/toy_flow.py) for a runnable example.

#### Example 2: VAE

Expand Down Expand Up @@ -87,6 +88,7 @@ model = Flow(base_dist=StandardNormal((latent_size,)),
VAE(encoder=encoder, decoder=decoder)
])
```
See [here](https://github.com/didriknielsen/survae_flows/blob/master/examples/mnist_vae.py) for a runnable example.

#### Example 3: Multi-Scale Augmented Flow

Expand Down Expand Up @@ -141,6 +143,7 @@ model = Flow(base_dist=StandardNormal((24,8,8)),
AffineCouplingBijection(net(24)), ActNormBijection2d(24), Conv1x1(24),
])
```
See [here](https://github.com/didriknielsen/survae_flows/blob/master/examples/cifar10_aug_flow.py) for a runnable example.


#### Acknowledgements
Expand Down

0 comments on commit 5d81a8e

Please sign in to comment.