Commit 1b3b50d0 authored by Yuxin Wu's avatar Yuxin Wu

use zeros_initializer..

parent f12517e5
......@@ -46,7 +46,7 @@ def get_global_step_var():
assert scope.name == '', \
"Creating global_step_var under a variable scope would cause problems!"
var = tf.get_variable(GLOBAL_STEP_OP_NAME, shape=[],
#initializer=tf.constant_initializer(value=0, dtype=tf.int32),
initializer=tf.zeros_initializer,
trainable=False, dtype=tf.int32)
return var
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment