variational_inference.py 文件源码

python
阅读 28 收藏 0 点赞 0 评论 0

项目:proximity_vi 作者: altosaar 项目源码 文件源码
def build_summary_op(self):
    cfg = self.config
    self.saver = tf.train.Saver(max_to_keep=5)
    self.summary_writer = tf.summary.FileWriter(
        cfg['log/dir'], self.session.graph, flush_secs=2)
    assert_op = tf.verify_tensor_all_finite(self.elbo_sum, 'ELBO check')
    with tf.control_dependencies([assert_op]):
      self.summary_op = tf.summary.merge_all()
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号