Загрузка параметров внедрения ADAM с поддержкой отладки.
Операция, которая загружает параметры оптимизации в HBM для встраивания. Должна предшествовать операция ConfigureTPUEmbeddingHost, которая устанавливает правильную конфигурацию таблицы внедрения. Например, эта операция используется для установки параметров, которые загружаются из контрольной точки перед выполнением цикла обучения.
Вложенные классы
сорт | LoadTPUEmbeddingADAMParametersGradAccumDebug.Options | Дополнительные атрибуты для LoadTPUEmbeddingADAMParametersGradAccumDebug |
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
статический LoadTPUEmbeddingADAMParametersGradAccumDebug.Options | конфигурация (строковая конфигурация) |
статическая нагрузкаTPUEmbeddingADAMParametersGradAccumDebug | create (Область видимости , Операнд <TFloat32> параметры, Операнд <TFloat32> импульсы, Операнд <TFloat32> скорости, Операнд <TFloat32> градиентАккумуляторы, Длинные numShards, Длинные shardId, Опции... опции) Фабричный метод для создания класса, обертывающего новую операцию LoadTPUEmbeddingADAMParametersGradAccumDebug. |
статический LoadTPUEmbeddingADAMParametersGradAccumDebug.Options | tableId (длинный tableId) |
статический LoadTPUEmbeddingADAMParametersGradAccumDebug.Options | tableName (строка tableName) |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
общедоступная статическая конфигурация LoadTPUEmbeddingADAMParametersGradAccumDebug.Options (конфигурация String)
public static LoadTPUEmbeddingADAMParametersGradAccumDebug create (область области действия , параметры операнда <TFloat32> , операнд <TFloat32> импульса, операнд <TFloat32> скорости, операнд <TFloat32> градиента Аккумуляторы, длинные numShards, длинный shardId, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию LoadTPUEmbeddingADAMParametersGradAccumDebug.
Параметры
объем | текущий объем |
---|---|
параметры | Значение параметров, используемых в алгоритме оптимизации ADAM. |
импульс | Значение импульсов, используемых в алгоритме оптимизации ADAM. |
скорости | Значение скоростей, используемых в алгоритме оптимизации ADAM. |
ГрадиентАккумуляторы | Значение градиента_аккумуляторов, используемое в алгоритме оптимизации ADAM. |
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр LoadTPUEmbeddingADAMParametersGradAccumDebug