Установка пошаговой скорости обучения в TensorFlow Estimator API - PullRequest
0 голосов
/ 07 июня 2018

Я хочу установить кусочную скорость обучения для объекта оценки с тензорным потоком, пока что я делаю вот так

boundaries = [100000, 200000, 300000]
values = [0.1, 0.05, 0.01, 0.001]
global_step = tf.train.get_global_step()
learning_rate = tf.train.piecewise_constant(global_step, boundaries, values)

, а затем создаю оценщик с этими параметрами модели.

Он выдает ошибку ValueError: None values not supported., так как значение глобального шага равно NONE.

С другой стороны, когда я создаю global_step = tf.Variable(0, trainable=False), создается тензор скорости обучения, но позже он терпит неудачу при вызове создания объекта оценщика.,Уже проверяли это: Как использовать затухающую скорость обучения с оценкой в ​​тензорном потоке?

1 Ответ

0 голосов
/ 07 июня 2018

Вместо этого используйте tf.train.get_or_create_global_step(), он создаст для вас глобальный шаг с ожидаемым определением.

...