tf.contrib.opt.ModelAverageCustomGetter

Класс custom_getter используется для выполнения следующих операций.

tf.contrib.opt.ModelAverageCustomGetter(
    worker_device
)

1. Изменение обучаемых переменных на локальную коллекцию и размещение их на устройстве рабочего узла
2. Генерация глобальных переменных Обратите внимание, что класс следует использовать с tf.replica_device_setter, чтобы глобальные переменные центра и глобальная переменная шага могли быть размещены на устройстве ps. Кроме того, используйте 'tf.compat.v1.get_variable' вместо 'tf.Variable', чтобы использовать этот пользовательский getter.

Например, ma_custom_getter = ModelAverageCustomGetter(worker_device) с tf.device( tf.compat.v1.train.replica_device_setter( worker_device=worker_device, ps_device="/job:ps/cpu:0", cluster=cluster)), tf.compat.v1.variable_scope('',custom_getter=ma_custom_getter): hid_w = tf.compat.v1.get_variable( initializer=tf.random.truncated_normal( [IMAGE_PIXELS * IMAGE_PIXELS, FLAGS.hidden_units], stddev=1.0 / IMAGE_PIXELS), name="hid_w") hid_b = tf.compat.v1.get_variable(initializer=tf.zeros([FLAGS.hidden_units]), name="hid_b")

Методы

__call__

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

__call__(
    getter, name, trainable, collections, *args, **kwargs
)

Вызов self как функции.