Модуль: tf.train.experimental

Общедоступный API для пространства имен tf._api.v2.train.experimental

Классы

class MaxShardSizePolicy: Политика, которая разбивает тензоры на фрагменты с максимальным размером фрагмента.

class PythonState: Миксин для помещения состояния Python в контрольной точке на основе объектов.

class ShardByTaskPolicy: Политика, которая разбивает тензоры на фрагменты на основе задания спецификации устройства.

class ShardableTensor: Обёртка для тензора, содержащая данные, необходимые для фрагментации.

class ShardingCallback: Функция обратного вызова фрагментации контрольной точки вместе с текстовым описанием.