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

Small bug in tf_utils/layers.py #6

Open pukkapies opened 7 years ago

pukkapies commented 7 years ago

In line 70 I think there is a small mistake, it should be: int(input_shape[2] * strides[2]), int(input_shape[3] * strides[3])] It's not a major bug, as it would throw an error if the output shape didn't work out correctly. I think it runs OK just because the input is square.