View source on GitHub |
Returns all variables created with trainable=True
.
tf.compat.v1.trainable_variables(
scope=None
)
Migrate to TF2
Not compatible with eager execution and tf.function
. In particular, Graph
collections are deprecated in TF2. Instead please create a tf.Module
container for all your model state, including variables.
You can then list all the trainable variables in your tf.Module
through the
trainable_variables
attribute.
Description
When passed trainable=True
, the Variable()
constructor automatically
adds new variables to the graph collection
GraphKeys.TRAINABLE_VARIABLES
. This convenience function returns the
contents of that collection.
Returns | |
---|---|
A list of Variable objects. |