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.

Возвращает
Список словарей шардов, содержащих тензоры. [ {checkpoint key: {slice_spec: tensor} } ]

© 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