The bug in computing shift KL-divergence in the body of tf.while_loop has been fixed. Image inputs have been removed from tf.while_loop arguments and are now accessed from the global context of class members.
The bug in computing shift KL-divergence in the body of tf.while_loop has been fixed. Image inputs have been removed from tf.while_loop arguments and are now accessed from the global context of class members.

aakhundov/tf-attend-infer-repeat