Spec-Zone.ru › TensorFlow 1.15

tf.contrib.framework.load_variable_slot_initializer

Загружает предварительно обученные слоты для многоклассовых моделей линейных моделей из контрольной точки.

tf.contrib.framework.load_variable_slot_initializer(
    ckpt_path, old_tensor_name, primary_partition_info, new_row_vocab_size,
    new_col_vocab_size, old_row_vocab_file=None, new_row_vocab_file=None,
    old_col_vocab_file=None, new_col_vocab_file=None, num_row_oov_buckets=0,
    num_col_oov_buckets=0, initializer=None, max_rows_in_memory=-1
)

Обёртка вокруг load_and_remap_matrix_initializer() , специализированная для загрузки слотов многоклассовых моделей (например, аккумуляторов оптимизатора) и их переназначения в соответствии с предоставленными файлами словаря. Дополнительные сведения см. в документации к load_and_remap_matrix_initializer(). Принимает variable_scope._PartitionInfo, представляющий разбиение первичного Variable слота. Это необходимо, так как создание аккумуляторов Variable игнорирует информацию о первичном охвате и разбиении.

Аргументы
ckpt_path Путь к контрольной точке TensorFlow (версия 2, TensorBundle) из которой будет загружена старая матрица Tensor .
old_tensor_name Имя 2-мерной Tensor для загрузки из контрольной точки.
primary_partition_info variable_scope._PartitionInfo , содержащий информацию о разбиении первичного Variable слота. Используется для вычисления смещения и перезаписи информации о разбиении, переданной в вызов _initialize.
new_row_vocab_size int , определяющий количество элементов в new_row_vocab_file. Если переназначение строк не требуется (не предоставлен словарь строк), это должно быть равно количеству строк для загрузки из старой матрицы (что теоретически может быть меньше, чем количество строк в старой матрице).
new_col_vocab_size int , определяющий количество элементов в new_col_vocab_file. Если переназначение столбцов не требуется (не предоставлен словарь столбцов), это должно быть равно количеству столбцов в старой матрице.
old_row_vocab_file Скалярный Tensor типа string , содержащий путь к файлу словаря старых строк. Может быть None, что означает отсутствие переназначения по оси строк.
new_row_vocab_file Скалярный Tensor типа string , содержащий путь к файлу словаря новых строк. Может быть None, что означает отсутствие переназначения по оси строк.
old_col_vocab_file Скалярный Tensor типа string , содержащий путь к файлу словаря старых столбцов. Может быть None, что означает отсутствие переназначения по оси столбцов.
new_col_vocab_file Скалярный Tensor типа string , содержащий путь к файлу словаря новых столбцов. Может быть None, что означает отсутствие переназначения по оси столбцов.
num_row_oov_buckets int , определяющий количество строк вне словаря для добавления. Должно быть >= 0.
num_col_oov_buckets int , определяющий количество столбцов вне словаря для добавления. Должно быть >= 0.
initializer Функция инициализатора для инициализации отсутствующих значений. Принимает 1-мерный тензор в качестве аргумента для указания формы возвращаемого тензора. Если None, по умолчанию используется zeros_initializer().
max_rows_in_memory int , определяющий максимальное количество строк для загрузки из контрольной точки за один раз. Если меньше или равно 0, вся матрица загрузится в память. Установка этого аргумента обменивает увеличение чтения с диска на меньшее использование памяти.
Возвращает
Функцию инициализации переменной, которая должна использоваться для инициализации (возможно, разбиеной) Variable , полная форма которой [new_row_vocab_size + num_row_oov_buckets, new_col_vocab_size + num_col_oov_buckets].
Исключения
TypeError Если initializer указано, но не является вызываемым.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/framework/load_variable_slot_initializer

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API