tf.contrib.eager.EagerVariableStore

Обёртка, позволяющая использовать функциональные слои с жадной (eager) обработкой.

tf.contrib.eager.EagerVariableStore(
    store=None
)

Когда включена жадная обработка, переменные удаляются при выходе из области видимости и по умолчанию не сохраняются в глобальных коллекциях. Много кода (в основном функциональные слои в tf.layers) предполагает, что переменные хранятся в глобальном списке.

EagerVariableStore может использоваться вместе с этим кодом для обеспечения поддержки жадной обработки. Например, для создания плотного слоя используйте:

container = tfe.EagerVariableStore()
for input in dataset_iterator:
  with container.as_default():
    x = tf.compat.v1.layers.dense(input, name="l1")
print(container.variables)  # Should print the variables used in the layer.

Методы

as_default

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

as_default()

copy

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

copy()

Копирование этого хранилища переменных и всех его содержимых.

Переменные, содержащиеся в этом хранилище, будут скопированы в новое хранилище переменных, что означает, что они могут быть изменены без влияния на переменные в этом хранилище.

non_trainable_variables

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

non_trainable_variables()

trainable_variables

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

trainable_variables()

variables

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

variables()