"models/vscode:/vscode.git/clone" did not exist on "acd87ae02adfd1fc08f37d367fd250f4f0d92003"
Commit c74897b0 authored by Sergio Guadarrama's avatar Sergio Guadarrama
Browse files

Merge pull request #42 from tensorflow/sguada-patch-1

Update ops.py
parents 044a5f5b dd0671b7
...@@ -76,7 +76,7 @@ def batch_norm(inputs, ...@@ -76,7 +76,7 @@ def batch_norm(inputs,
""" """
inputs_shape = inputs.get_shape() inputs_shape = inputs.get_shape()
with tf.variable_op_scope([inputs], scope, 'BatchNorm', reuse=reuse): with tf.variable_op_scope([inputs], scope, 'BatchNorm', reuse=reuse):
axis = range(len(inputs_shape) - 1) axis = list(range(len(inputs_shape) - 1))
params_shape = inputs_shape[-1:] params_shape = inputs_shape[-1:]
with scopes.arg_scope([variables.variable], restore=restore): with scopes.arg_scope([variables.variable], restore=restore):
# Allocate parameters for the beta and gamma of the normalization. # Allocate parameters for the beta and gamma of the normalization.
......
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