Skip to content

Instantly share code, notes, and snippets.

@ericjang

ericjang/nf2_chain.py

Created January 17, 2018 18:25
Show Gist options
  • Save ericjang/b30f88227d531a8e02ff4b5223bce406 to your computer and use it in GitHub Desktop.
Save ericjang/b30f88227d531a8e02ff4b5223bce406 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
nf2_chain.py
for i in range(num_bijectors):
bijectors.append(tfb.MaskedAutoregressiveFlow(
shift_and_log_scale_fn=tfb.masked_autoregressive_default_template(
hidden_layers=[512, 512])))
bijectors.append(tfb.Permute(permutation=[1, 0]))
flow_bijector = tfb.Chain(list(reversed(bijectors[:-1])))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment