Skip to content

GitLab

Commit 0e5cc640 authored by Olivia's avatar Olivia Committed by GitHub
Browse files

update the initializer changes

parent f565b808
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
inception/inception/slim/ops.py
View file @ 0e5cc640
...@@ -85,7 +85,7 @@ def batch_norm(inputs, ...@@ -85,7 +85,7 @@ def batch_norm(inputs,
if center: if center:
beta = variables.variable('beta', beta = variables.variable('beta',
params_shape, params_shape,
initializer=tf.zeros_initializer, initializer=tf.zeros_initializer(),
trainable=trainable, trainable=trainable,
restore=restore) restore=restore)
if scale: if scale:
...@@ -99,7 +99,7 @@ def batch_norm(inputs, ...@@ -99,7 +99,7 @@ def batch_norm(inputs,
moving_collections = [moving_vars, tf.GraphKeys.MOVING_AVERAGE_VARIABLES] moving_collections = [moving_vars, tf.GraphKeys.MOVING_AVERAGE_VARIABLES]
moving_mean = variables.variable('moving_mean', moving_mean = variables.variable('moving_mean',
params_shape, params_shape,
initializer=tf.zeros_initializer, initializer=tf.zeros_initializer(),
trainable=False, trainable=False,
restore=restore, restore=restore,
collections=moving_collections) collections=moving_collections)
......
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