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

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

Классы

class DynamicLossScale: Масштабирование потерь, которое динамически подстраивается.

class FixedLossScale: Масштабирование потерь с фиксированным значением.

class LossScale: Базовый класс для всех масштабирований потерь TF1.

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

class MixedPrecisionLossScaleOptimizer: Оптимизатор, применяющий масштабирование потерь.

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

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

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

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

Функции

disable_mixed_precision_graph_rewrite(...): Отключение перестройки графа смешанной точности.

enable_mixed_precision_graph_rewrite(...): Включение смешанной точности с помощью перестройки графа.