tf.compat.v1.train.Saver

Сохраняет и восстанавливает переменные.

tf.compat.v1.train.Saver(
    var_list=None, reshape=False, sharded=False, max_to_keep=5,
    keep_checkpoint_every_n_hours=10000.0, name=None, restore_sequentially=False,
    saver_def=None, builder=None, defer_build=False, allow_empty=False,
    write_version=tf.train.SaverDef.V2, pad_step_number=False,
    save_relative_paths=False, filename=None
)

См. Переменные для обзора переменных, сохранения и восстановления.

Класс Saver добавляет операции для сохранения и восстановления переменных из и в контрольные точки. Он также предоставляет удобные методы для выполнения этих операций.

Контрольные точки — это двоичные файлы в собственном формате, которые отображают имена переменных на значения тензоров. Лучший способ изучить содержимое контрольной точки — загрузить её с помощью Saver.

Сохранители могут автоматически нумеровать имена файлов контрольных точек с помощью предоставленного счётчика. Это позволяет сохранять несколько контрольных точек на разных шагах во время обучения модели. Например, можно нумеровать имена файлов контрольных точек с номером шага обучения. Чтобы избежать заполнения дисков, сохранители автоматически управляют файлами контрольных точек. Например, они могут сохранять только N последних файлов или одну контрольную точку каждые N часов обучения.

Вы нумеруете имена файлов контрольных точек, передавая значение в необязательный аргумент global_step в save():

saver.save(sess, 'my-model', global_step=0) ==> filename: 'my-model-0'
...
saver.save(sess, 'my-model', global_step=1000) ==> filename: 'my-model-1000'

Кроме того, необязательные аргументы конструктора Saver() позволяют вам контролировать распространение файлов контрольных точек на диске:

• max_to_keep указывает максимальное количество последних файлов контрольных точек, которые нужно сохранить. По мере создания новых файлов старые файлы удаляются. Если None или 0, контрольные точки не удаляются из файловой системы, но только последняя сохраняется в файле checkpoint. По умолчанию значение 5 (то есть сохраняются 5 последних файлов контрольных точек).

• keep_checkpoint_every_n_hours: В дополнение к сохранению последних max_to_keep файлов контрольных точек, вы можете сохранить один файл контрольной точки каждые N часов обучения. Это может быть полезно, если вы хотите позже проанализировать, как модель развивалась в течение длительной сессии обучения. Например, передача keep_checkpoint_every_n_hours=2 гарантирует, что вы сохраняете один файл контрольной точки каждые 2 часа обучения. Значение по умолчанию 10 000 часов фактически отключает эту функцию.

Обратите внимание, что вам по-прежнему нужно вызвать метод save() для сохранения модели. Передача этих аргументов в конструктор не будет автоматически сохранять переменные.

Программа обучения, которая регулярно сохраняет данные, выглядит следующим образом:

...
# Create a saver.
saver = tf.compat.v1.train.Saver(...variables...)
# Launch the graph and train, saving the model every 1,000 steps.
sess = tf.compat.v1.Session()
for step in xrange(1000000):
    sess.run(..training_op..)
    if step % 1000 == 0:
        # Append the step number to the checkpoint name:
        saver.save(sess, 'my-model', global_step=step)

В дополнение к файлам контрольных точек, сохранители сохраняют протокол буфера на диске со списком последних контрольных точек. Это используется для управления пронумерованными файлами контрольных точек и latest_checkpoint(), что упрощает поиск пути к последней контрольной точке. Этот протокол буфера хранится в файле с именем «checkpoint» рядом с файлами контрольных точек.

Если вы создаёте несколько сохранителей, вы можете указать другое имя файла для файла протокола буфера в вызове save().

Методы

as_saver_def

Просмотреть исходный код

as_saver_def()

Генерирует представление SaverDef этого сохранителя.

build

Просмотреть исходный код

build()

export_meta_graph

Просмотреть исходный код

export_meta_graph(
    filename=None, collection_list=None, as_text=False, export_scope=None,
    clear_devices=False, clear_extraneous_savers=False, strip_default_attrs=False,
    save_debug_info=False
)

Записывает MetaGraphDef в save_path/filename.

from_proto

Просмотреть исходный код

@staticmethod
from_proto(
    saver_def, import_scope=None
)

Возвращает объект Saver, созданный из saver_def.

recover_last_checkpoints

Просмотреть исходный код

recover_last_checkpoints(
    checkpoint_paths
)

Восстанавливает внутреннее состояние сохранителя после сбоя.

Этот метод полезен для восстановления состояния «self._last_checkpoints».

Ищет контрольные точки, указанные в checkpoint_paths. Если файлы существуют, использует их mtime как временную метку контрольной точки.

restore

Просмотр исходного кода

restore(
    sess, save_path
)

Восстанавливает ранее сохранённые переменные.

Этот метод выполняет операции, добавленные конструктором для восстановления переменных. Он требует сессии, в которой был запущен граф. Переменные для восстановления не обязательно должны быть инициализированы, так как восстановление само по себе является способом инициализации переменных.

Аргумент save_path обычно является значением, ранее возвращённым методом save(), или вызовом latest_checkpoint().

save

Просмотр исходного кода

save(
    sess, save_path, global_step=None, latest_filename=None,
    meta_graph_suffix='meta', write_meta_graph=True, write_state=True,
    strip_default_attrs=False, save_debug_info=False
)

Сохраняет переменные.

Этот метод выполняет операции, добавленные конструктором для сохранения переменных. Он требует сессии, в которой был запущен граф. Переменные для сохранения также должны быть инициализированы.

Метод возвращает префикс пути к недавно созданным файлам контрольного пункта. Эта строка может быть передана непосредственно в вызов restore().

set_last_checkpoints

Просмотр исходного кода

set_last_checkpoints(
    last_checkpoints
)

УСТАРЕЛО: Используйте set_last_checkpoints_with_time.

Устанавливает список старых имён файлов контрольных пунктов.

set_last_checkpoints_with_time

Просмотр исходного кода

set_last_checkpoints_with_time(
    last_checkpoints_with_time
)

Устанавливает список старых имён файлов контрольных пунктов и временных меток.

to_proto

Просмотр исходного кода

to_proto(
    export_scope=None
)

Преобразует этот Saver в протобуфер SaverDef.