TensorFlow

API

 tf.compat / v1 / v1.train.get_global_step


Small helper to get the global step.

# Create a variable to hold the global_step.
global_step_tensor = tf.Variable(10, trainable=False, name='global_step')
# Create a session.
sess = tf.compat.v1.Session()
# Initialize the variable
sess.run(global_step_tensor.initializer)
# Get the variable value.
print('global_step: %s' % tf.compat.v1.train.global_step(sess,
global_step_tensor))

global_step: 10

sess A TensorFlow Session object.
global_step_tensor Tensor or the name of the operation that contains the global step.

The global step value.

此页内容是否对您有帮助