tf.train.experimental.MaxShardSizePolicy

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

Наследуется от: ShardingCallback

Псевдонимы

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.train.experimental.MaxShardSizePolicy

tf.train.experimental.MaxShardSizePolicy(
    max_shard_size: int
)

Фрагменты могут превышать максимальный размер фрагмента, если они содержат 1. одиночный скалярный/строковый тензор, который нельзя было разделить и который превышает максимальный размер фрагмента, или 2. граф объекта контрольной точки, размер которого нельзя рассчитать при сохранении.

Атрибуты
`description`

Методы

`call`

Просмотреть исходный код

__call__(
    shardable_tensors: Sequence[tf.train.experimental.ShardableTensor]
) -> Sequence[sharding_util.TensorSliceDict]

Обработчик для разделения тензоров на фрагменты с максимальным размером фрагмента.

Аргументы
`shardable_tensors`	Список ShardableTensors.

Возвращает
Список словарей фрагментов, содержащих тензоры. [ {ключ контрольной точки: {спецификация среза: тензор} } ]

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/train/experimental/MaxShardSizePolicy