UnboundLocalError: local variable 'out' referenced before assignment

[wenwei202]

I run python plot_parametric_plot.py -n C1, and get following error:

Traceback (most recent call last):
  File "plot_parametric_plot.py", line 64, in <module>
    model = network_zoo.shallownet(nb_classes)
  File "/home//github/users/wenwei202/large-batch-training/network_zoo.py", line 37, in shallownet
    model.add(BatchNormalization(mode=2,axis=1))
  File "/home//anaconda2/lib/python2.7/site-packages/Keras-1.0.0-py2.7.egg/keras/models.py", line 139, in add
    output_tensor = layer(self.outputs[0])
  File "/home//anaconda2/lib/python2.7/site-packages/Keras-1.0.0-py2.7.egg/keras/engine/topology.py", line 485, in __call__
    self.add_inbound_node(inbound_layers, node_indices, tensor_indices)
  File "/home//anaconda2/lib/python2.7/site-packages/Keras-1.0.0-py2.7.egg/keras/engine/topology.py", line 543, in add_inbound_node
    Node.create_node(self, inbound_layers, node_indices, tensor_indices)
  File "/home//anaconda2/lib/python2.7/site-packages/Keras-1.0.0-py2.7.egg/keras/engine/topology.py", line 148, in create_node
    output_tensors = to_list(outbound_layer.call(input_tensors[0], mask=input_masks[0]))
  File "/home//anaconda2/lib/python2.7/site-packages/Keras-1.0.0-py2.7.egg/keras/layers/normalization.py", line 118, in call
    return out
UnboundLocalError: local variable 'out' referenced before assignment

Keras version: 1
Tensorflow: '1.4.0'