tf.compat.v1.fixed_size_partitioner

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

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

Миграция на TF2

Этот API устарел в TF2. В TF2 разделитель больше не является частью объявления переменной через tf.Variable. Обучение с параметрическим сервером обрабатывает разделение переменных. Соответствующий класс разделителя 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