Get the global step tensor.
tf.compat.v1.train.get_global_step( graph=None )
The global step tensor must be an integer variable. We first try to find it in the collection GLOBAL_STEP
, or by name global_step:0
.
Args | |
---|---|
graph | The graph to find the global step in. If missing, use default graph. |
Returns | |
---|---|
The global step variable, or None if none was found. |
Raises | |
---|---|
TypeError | If the global step tensor has a non-integer type, or if it is not a Variable . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/train/get_global_step