Операция, которая извлекает параметры оптимизации из встраивания в память хоста.
Операция, которая извлекает параметры оптимизации из встраивания в память хоста. Должна предшествовать операция ConfigureTPUEmbeddingHost, которая устанавливает правильную конфигурацию таблицы внедрения. Например, эта операция используется для получения обновленных параметров перед сохранением контрольной точки. Для Адаграда вспомогательный1 будет содержать аккумуляторы после выполнения этой операции. Для SGD все вспомогательные* значения будут пустыми (тензоры 0x0 для этой таблицы). Для FTRL вспомогательный1 будет содержать аккумуляторы, а вспомогательный2 — линейные члены. Для ADAM вспомогательный1 будет содержать импульсы, а вспомогательный2 — скорости.
Публичные методы
| Список< Вывод <Float>> | вспомогательный1 () Список тензоров, по одному для каждой таблицы внедрения, содержащий первый сохраненный вспомогательный параметр оптимизации. |
| Список< Вывод <Float>> | вспомогательный2 () Список тензоров, по одному для каждой таблицы внедрения, содержащий второй сохраненный вспомогательный параметр оптимизации. |
| Список< Вывод <Float>> | вспомогательный3 () Список тензоров, по одному для каждой таблицы внедрения, содержащий третий сохраненный вспомогательный параметр оптимизации. |
| Список< Вывод <Float>> | вспомогательный4 () Список тензоров, по одному для каждой таблицы внедрения, содержащий четвертый сохраненный вспомогательный параметр оптимизации. |
| Список< Вывод <Float>> | вспомогательный5 () Список тензоров, по одному для каждой таблицы внедрения, содержащий пятый сохраненный вспомогательный параметр оптимизации. |
| Список< Вывод <Float>> | вспомогательный6 () Список тензоров, по одному для каждой таблицы внедрения, содержащий шесть сохраненных вспомогательных параметров оптимизации. |
| Список< Вывод <Float>> | вспомогательный7 () Список тензоров, по одному для каждой таблицы внедрения, содержащий сохраненный седьмой вспомогательный параметр оптимизации. |
| статический RetriveAllTPUEmbeddingParameters | создать (область области действия , длинные NumTables, конфигурация строки, длинные numShards, длинный shardId) Фабричный метод для создания класса, обертывающего новую операцию RetrieveAllTPUEmbeddingParameters. |
| Список< Вывод <Float>> | параметры () Список тензоров, по одному для каждой таблицы внедрения, содержащий сохраненные параметры таблицы внедрения. |
Унаследованные методы
Публичные методы
общедоступный список< Вывод <Float>> вспомогательный1 ()
Список тензоров, по одному для каждой таблицы внедрения, содержащий первый сохраненный вспомогательный параметр оптимизации. Элементы присутствуют в списке, но имеют нулевой размер для неиспользуемых параметров оптимизации (в зависимости от алгоритма, используемого для каждой таблицы).
общедоступный список < Вывод <Float>> вспомогательный2 ()
Список тензоров, по одному для каждой таблицы внедрения, содержащий второй сохраненный вспомогательный параметр оптимизации. Элементы присутствуют в списке, но имеют нулевой размер для неиспользуемых параметров оптимизации (в зависимости от алгоритма, используемого для каждой таблицы).
общедоступный список < Вывод <Float>> вспомогательный3 ()
Список тензоров, по одному для каждой таблицы внедрения, содержащий третий сохраненный вспомогательный параметр оптимизации. Элементы присутствуют в списке, но имеют нулевой размер для неиспользуемых параметров оптимизации (в зависимости от алгоритма, используемого для каждой таблицы).
общедоступный список < Вывод <Float>> вспомогательный4 ()
Список тензоров, по одному для каждой таблицы внедрения, содержащий четвертый сохраненный вспомогательный параметр оптимизации. Элементы присутствуют в списке, но имеют нулевой размер для неиспользуемых параметров оптимизации (в зависимости от алгоритма, используемого для каждой таблицы).
общедоступный список < Вывод <Float>> вспомогательный5 ()
Список тензоров, по одному для каждой таблицы внедрения, содержащий пятый сохраненный вспомогательный параметр оптимизации. Элементы присутствуют в списке, но имеют нулевой размер для неиспользуемых параметров оптимизации (в зависимости от алгоритма, используемого для каждой таблицы).
общедоступный список < Вывод <Float>> вспомогательный6 ()
Список тензоров, по одному для каждой таблицы внедрения, содержащий шесть сохраненных вспомогательных параметров оптимизации. Элементы присутствуют в списке, но имеют нулевой размер для неиспользуемых параметров оптимизации (в зависимости от алгоритма, используемого для каждой таблицы).
общедоступный список< Вывод <Float>> вспомогательный7 ()
Список тензоров, по одному для каждой таблицы внедрения, содержащий сохраненный седьмой вспомогательный параметр оптимизации. Элементы присутствуют в списке, но имеют нулевой размер для неиспользуемых параметров оптимизации (в зависимости от алгоритма, используемого для каждой таблицы).
public static RetrieveAllTPUEmbeddingParameters create (область области действия , длинные NumTables, конфигурация строки, длинные numShards, длинный shardId)
Фабричный метод для создания класса, обертывающего новую операцию RetrieveAllTPUEmbeddingParameters.
Параметры
| объем | текущий объем |
|---|---|
| NumTables | Количество встраиваемых таблиц. |
| конфигурация | Прототип TPUEmbeddingConfiguration, описывающий загружаемые параметры таблицы, сериализованные в строку. |
| число осколков | Количество сегментов, на которые разделены таблицы внедрения. |
| идентификатор осколка | Идентификатор шарда для этой операции. |
Возврат
- новый экземпляр RetrieveAllTPUEmbeddingParameters
общедоступный список <Выходные параметры <Float> > ()
Список тензоров, по одному для каждой таблицы внедрения, содержащий сохраненные параметры таблицы внедрения.