tf.compat.v1.fixed_size_partitioner

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

tf.compat.v1.fixed_size_partitioner(
    num_shards, axis=0
)

Переход на TF2

В TF2 этот API устарел. В TF2 разделитель больше не является частью объявления переменной с помощью tf.Variable. Обучение с ParameterServer выполняет сегментирование переменных. Соответствующий класс разделителя TF2 для fixed_size_partitioner — tf.distribute.experimental.partitioners.FixedShardsPartitioner.

См. руководство по переходу, чтобы узнать о различиях в обработке переменных и потерь в TF1 и TF2.

До перехода:

x = tf.compat.v1.get_variable(
  "x", shape=(2,), partitioner=tf.compat.v1.fixed_size_partitioner(2)
)

После перехода:

partitioner = (
    tf.distribute.experimental.partitioners.FixedShardsPartitioner(
        num_shards=2)
)
strategy = tf.distribute.experimental.ParameterServerStrategy(
               cluster_resolver=cluster_resolver,
               variable_partitioner=partitioner)

with strategy.scope():
  x = tf.Variable([1.0, 2.0])

Описание

Аргументы
`num_shards`	`int`, количество сегментов для разделения переменной.
`axis`	`int`, ось для разделения.

Возвращает
Функцию разделения, которую можно использовать в качестве аргумента `partitioner` для `variable_scope` и `get_variable`.

© 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/compat/v1/fixed_size_partitioner