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