https://gist.github.com/Wermarter/466e9585579ef65927fa934fe4e0ffd4 在这里,我尝试使用TFLearn在Tensorflow中实现Variational AutoEncoder。
我在self.training_model.session
的一个大图中构建了训练,编码和生成的计算。 self.generating_model
和self.recognition_model
与self.training_model
共享同一会话。
当我运行generating_model
以生成MNIST 2D Latent空间时,一切都很顺利。但是当我运行self.recognition_model
来编码给定的input_data时出现错误,它要求我给属于self.train_data
的{{1}}输入值。
这里有完整的错误:
self.training_model
答案 0 :(得分:0)
这是特定于代码的错误。我的self.recognition_model
实际上与self.train_data
中的占位符self.curr_batch_size
到self._sample_z()
相关联。我的解决方案是将self.curr_batch_size
重新关联到self.input_data
的大小。
那就是它。快乐编码