在tensorflow中,是否可以在使用sess.run()执行的程序部分中使用步骤号(例如,通过提供它,例如通过feed_dict
发布于 2016-12-13 17:43:59
是。要创建全局步骤,可以执行以下任一操作:
step = tf.contrib.framework.create_global_step()
global_step_tensor = tf.Variable(10,trainable=False,tf.train.global_step=‘global_step’) tf.train.global_step(sess,global
要检索全局步骤,可以使用tf.contrib.framework.get_global_step (TF0.11)或tf.train.get_global_step (TF0.12)。它们返回一个变量,您可以自由访问/赋值:
sess.run(tf.contrib.framework.get_global_step())
Out: 0
step_place = tf.placeholder(tf.int64)
ass_op = tf.assign(tf.contrib.framework.get_global_step(), step_place)
sess.run(ass_op, feed_dict={step_place: 42})
Out: 42
sess.run(tf.contrib.framework.get_global_step())
Out: 42https://stackoverflow.com/questions/41115950
复制相似问题