tf.contrib.opt.AGNCustomGetter

Класс custom_getter используется для:

tf.contrib.opt.AGNCustomGetter(
    worker_device
)

1. Изменение обучаемых переменных на локальную коллекцию и размещение их на устройстве рабочего узла
2. Генерация глобальных переменных (глобальные переменные центра)
3. Генерация переменных градиента (градиенты), которые записывают сумму градиентов и размещают их на устройстве рабочего узла. Обратите внимание, что класс должен использоваться с tf.replica_device_setter, чтобы глобальные переменные центра и глобальная переменная шага могли быть размещены на устройстве ps.

Методы

__call__

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

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

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