tf.contrib.cloud.ConfigureGcsHook

ConfigureGcsHook настраивает GCS при использовании с Estimator/TPUEstimator.

Наследуется от: SessionRunHook

tf.contrib.cloud.ConfigureGcsHook(
    credentials=None, block_cache=None
)

Пример:

sess = tf.compat.v1.Session()
refresh_token = raw_input("Refresh token: ")
client_secret = raw_input("Client secret: ")
client_id = "<REDACTED>"
creds = {
    "client_id": client_id,
    "refresh_token": refresh_token,
    "client_secret": client_secret,
    "type": "authorized_user",
}
tf.contrib.cloud.configure_gcs(sess, credentials=creds)

Методы

after_create_session

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

after_create_session(
    session, coord
)

Вызывается при создании новой сессии TensorFlow.

Это вызывается, чтобы сообщить хукам, что была создана новая сессия. Это имеет два существенных отличия от ситуации, в которой вызывается begin:

• Когда это вызывается, граф окончательно сформирован, и к нему больше нельзя добавлять операции.
• Этот метод также будет вызван в результате восстановления обернутой сессии, а не только в начале всей сессии.

after_run

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

after_run(
    run_context, run_values
)

Вызывается после каждого вызова run().

Аргумент run_values содержит результаты запрошенных операций/тензоров вызовом before_run().

Аргумент run_context — тот же, что и переданный в вызов before_run. run_context.request_stop() может быть вызван для остановки итерации.

Если session.run() вызывает какие-либо исключения, то after_run() не вызывается.

before_run

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

before_run(
    run_context
)

Вызывается перед каждым вызовом run().

Вы можете вернуть из этого вызова объект SessionRunArgs, указывающий операции или тензоры для добавления к предстоящему вызову run(). Эти операции/тензоры будут выполнены вместе с операциями/тензорами, изначально переданными в исходный вызов run(). Аргументы вызова run, которые вы возвращаете, также могут содержать значения feed, которые нужно добавить к вызову run().

Аргумент run_context — SessionRunContext, который предоставляет информацию о предстоящем вызове run(): исходно запрошенные операции/тензоры, сессия TensorFlow.

На этом этапе граф окончательно сформирован, и к нему нельзя больше добавлять операции.

begin

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

begin()

Вызывается один раз перед использованием сессии.

При вызове текущий граф — тот, который будет запущен в сессии. Хук может изменять граф, добавляя новые операции. После вызова begin() граф будет окончательно сформирован, и другие обратные вызовы больше не смогут изменять граф. Второй вызов begin() на том же графе не должен изменять граф.

end

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

end(
    session
)

Вызывается в конце сессии.

Аргумент session можно использовать в случае, если хук хочет выполнить окончательные операции, такие как сохранение последней контрольной точки.

Если session.run() вызывает исключение, отличное от OutOfRangeError или StopIteration, то end() не вызывается. Обратите внимание на разницу между поведением end() и after_run() при возникновении session.run() исключений OutOfRangeError или StopIteration. В этом случае end() вызывается, но after_run() — нет.