caicloud / tensorflow-tutorial

Example TensorFlow codes and Caicloud TensorFlow as a Service dev environment.
2.93k stars 2.08k forks source link

Base 框架添加关闭自动计算 summary 机制的方法;MNIST 样例自定义初始化函数添加接收 tf.train.Scaffold 对象参数。 #20

Closed lienhua34 closed 7 years ago

lienhua34 commented 7 years ago

Base 框架 PR https://github.com/caicloud/clever-admin/pull/353

  1. 处理 MNIST 样例自定义初始化异常问题。

    Base 框架调整为采用 tf.train.MonitoredTrainingSession 来管理分布式 Session 后,自定义初始化函数 init_fn 必须要接收两个参数:tf.train.Scaffold 对象和 tf.Session 对象。

  2. Base 框架的 ModelFnHandler 对象添加 summary_op 参数用于自定义计算 summary 的操作或者关闭自动计算 summary 机制。

    默认情况下,Base 框架会通过 tf.summary.merge_all() 函数来计算并收集模型 Graph 中的所有 summary 信息。但是,当某些 summary 信息的计算需要对 placeholder 的 feed 时,Base 框架会发生异常。

    于是通过 ModelFnHandler 对象的 summary_op 参数来让用户提供自动计算 summary 的操作,或者将该参数设置为 None 来关闭自动计算模型 summary 机制。

lienhua34 commented 7 years ago

@perhapszzy PTAL