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

Публичный API пространства имён tf._api.v2.train.experimental

Классы

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

class PythonState: Миксин для включения состояния Python в контрольную точку на основе объектов.

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

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

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