权重初始化Tensorflow tf.estimator

时间:2018-09-27 08:57:41

标签: tensorflow initialization

有没有办法在预建的tf.estimator中调整权重初始化? 我想使用Xavier('a')之后或He中的方法。默认使用哪种方法?我无法从文档中弄清楚。

我使用DNNRegressor。

2 个答案:

答案 0 :(得分:2)

DNNRegressor使用glorot_uniform_initializer(又名Xavier制服),它被硬编码在implementation中。

要将不同的初始化器与estimators API配合使用,您必须使用custom estimator

答案 1 :(得分:0)

自定义估算器要求您从头开始定义图,您可以为每个图层/操作传递初始化器(如果其构造函数带有初始化器参数),通常在model_fn或定义图的函数内部完成