tf.contrib.checkpoint.capture_dependencies

View source on GitHub

Capture variables created within this scope as Template dependencies.

@tf_contextlib.contextmanager
tf.contrib.checkpoint.capture_dependencies(
    template
)

Requires that template.variable_scope is active.

This scope is intended as a compatibility measure, allowing a trackable object to add dependencies on variables created in a block of code which is not aware of object-based saving (and instead uses variable names heavily). This is how Template objects add dependencies on variables and sub-Templates. Where possible, use tf.compat.v1.make_template directly.

template The Template object to register dependencies with.

Yields:

None (when used as a context manager).