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