权重初始化 Tensorflow tf.estimator.



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

我使用 DNNRegressor。

DNNRegressor使用glorot_uniform_initializer(又名Xavier制服(,它在实现中是硬编码的。

要将不同的初始值设定项与估算器 API 一起使用,您必须使用自定义估算器。

自定义估计器要求您从头开始定义图形,您可以为每个层/操作传递初始值设定项(如果其构造函数采用初始值设定项参数(,它通常在model_fn内部完成,或者在定义图形的函数中完成

最新更新