tf.train.Saver

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

tf.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.