Closed lienhua34 closed 7 years ago
Base 框架 PR https://github.com/caicloud/clever-admin/pull/353
处理 MNIST 样例自定义初始化异常问题。
Base 框架调整为采用 tf.train.MonitoredTrainingSession 来管理分布式 Session 后,自定义初始化函数 init_fn 必须要接收两个参数:tf.train.Scaffold 对象和 tf.Session 对象。
Base 框架的 ModelFnHandler 对象添加 summary_op 参数用于自定义计算 summary 的操作或者关闭自动计算 summary 机制。
默认情况下,Base 框架会通过 tf.summary.merge_all() 函数来计算并收集模型 Graph 中的所有 summary 信息。但是,当某些 summary 信息的计算需要对 placeholder 的 feed 时,Base 框架会发生异常。
于是通过 ModelFnHandler 对象的 summary_op 参数来让用户提供自动计算 summary 的操作,或者将该参数设置为 None 来关闭自动计算模型 summary 机制。
@perhapszzy PTAL
Base 框架 PR https://github.com/caicloud/clever-admin/pull/353
处理 MNIST 样例自定义初始化异常问题。
Base 框架调整为采用 tf.train.MonitoredTrainingSession 来管理分布式 Session 后,自定义初始化函数 init_fn 必须要接收两个参数:tf.train.Scaffold 对象和 tf.Session 对象。
Base 框架的 ModelFnHandler 对象添加 summary_op 参数用于自定义计算 summary 的操作或者关闭自动计算 summary 机制。
默认情况下,Base 框架会通过 tf.summary.merge_all() 函数来计算并收集模型 Graph 中的所有 summary 信息。但是,当某些 summary 信息的计算需要对 placeholder 的 feed 时,Base 框架会发生异常。
于是通过 ModelFnHandler 对象的 summary_op 参数来让用户提供自动计算 summary 的操作,或者将该参数设置为 None 来关闭自动计算模型 summary 机制。