search

openai / iaf

Code for reproducing key results in the paper "Improving Variational Inference with Inverse Autoregressive Flow"
https://arxiv.org/abs/1606.04934
MIT License
518 stars 133 forks source link

Incorrect initialisation in tf_utils/layers.py #7

Open pukkapies opened 7 years ago

pukkapies commented 7 years ago

In line 93 of the above file, it looks like the normalisation is performed over the out_channels, instead of in_channels. I think instead it should be: v_norm = tf.nn.l2_normalize(v, [0, 1, 3])