Инициализация веса Tensorflow tf.estimator

Question

Есть ли способ настроить инициализацию веса в предварительно собранном tf.estimator?Я хотел бы использовать метод после Ксавье (tf.contrib.layers.xavier_initializer) или от He.Какой метод используется по умолчанию?Я не мог понять это из документации.

Я использую DNNRegressor.

Answer 1

Пользовательские оценщики требуют, чтобы вы определяли свой график с нуля, вы можете передать инициализатор для каждого слоя / операции (если его конструктор принимает аргумент инициализатора), это обычно делается внутри model_fn или внутри функции, которая определяет график

Answer 2

DNNRegressor использует glorot_uniform_initializer (он же форма Ксавье), он жестко закодирован в реализации .

Чтобы использовать другой инициализатор с API оценки, вам необходимоиспользовать пользовательский оценщик .