Как отключить сводку для Tensorflow Estimator?

Question

Я использую Tensorflow-GPU 1.8 API в Windows 10. Для многих проектов я использую tf.Estimator, который действительно отлично работает. Он выполняет несколько шагов, включая написание резюме для Tensorboard. Но сейчас файл 'events.out.tfevents' становится все больше, и я сталкиваюсь с ошибками "нехватки места". По этой причине я хочу отключить запись резюме или, по крайней мере, уменьшить количество написанных резюме.

Продолжая эту миссию, я узнал о RunConfig, который вы можете пропустить при создании tf.Estimator. Очевидно, что параметр «save_summary_steps» (по умолчанию 200) контролирует способ составления резюме. К сожалению, изменение этого параметра, похоже, не имеет никакого эффекта. Он не отключит (используя значение «Нет») сводку или уменьшит (выбирая более высокие значения, например, 3000) размер файла «events.out.tfevents».

Я надеюсь, что вы, ребята, можете помочь мне здесь. Любая помощь приветствуется.

Ура, Tobs.

Answer 1

Я наблюдал следующее поведение. Это не имеет смысла для меня, поэтому я надеюсь, что мы получим лучший ответ:

Когда input_fn получает данные от tf.data.TFRecordDataset, тогда количество шагов между событиями сохранения составляет минимум save_summary_steps и ( количество обучающих примеров , деленное на размер партии ). Это означает, что он делает это минимум раз в эпоху.

Когда input_fn получает данные от tf.TextLineReader, это следует за save_summary_steps, как и следовало ожидать, и я могу дать ему большое значение для нечастых обновлений.