Модуль: 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(...): Включает смешанную точность с помощью преобразования графа.