【发布时间】:2019-05-23 01:30:27
【问题描述】:
有没有办法在预建的 tf.estimator 中调整权重初始化?
我想在 Xavier (tf.contrib.layers.xavier_initializer) 或 He 之后使用该方法。默认使用哪种方法?我无法从文档中弄清楚。
我使用 DNNRegressor。
【问题讨论】:
有没有办法在预建的 tf.estimator 中调整权重初始化?
我想在 Xavier (tf.contrib.layers.xavier_initializer) 或 He 之后使用该方法。默认使用哪种方法?我无法从文档中弄清楚。
我使用 DNNRegressor。
【问题讨论】:
DNNRegressor 使用 glorot_uniform_initializer(又名 Xavier 制服),它被硬编码在 implementation 中。
要在估算器 API 中使用不同的初始化程序,您必须使用 custom estimator。
【讨论】:
tf.estimator.LinearRegressor 没有强调它使用哪种方法来初始化权重!有没有办法找到它或打印它。有什么建议吗?
自定义估算器要求您从头开始定义图形,您可以为每个层/操作传递初始化程序(如果其构造函数采用初始化程序参数),通常在 model_fn 或定义图形的函数内部完成
【讨论】: