keskarnitish / large-batch-training Goto Github PK

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

Traceback (most recent call last):
  File "plot_parametric_plot.py", line 64, in <module>
    model = network_zoo.shallownet(nb_classes)
  File "/home/nqluo/experiement/large-batch-training-master/network_zoo.py", line 37, in shallownet
    model.add(BatchNormalization(mode=2,axis=1))
  File "/home/nqluo/anaconda3/envs/tf14-gpu/lib/python3.7/site-packages/keras/legacy/interfaces.py", line 34, in wrapper
    args, kwargs, converted = preprocessor(args, kwargs)
  File "/home/nqluo/anaconda3/envs/tf14-gpu/lib/python3.7/site-packages/keras/legacy/interfaces.py", line 451, in batchnorm_args_preprocessor
    raise TypeError('The `mode` argument of `BatchNormalization` '
TypeError: The `mode` argument of `BatchNormalization` no longer exists. `mode=1` and `mode=2` are no longer supported.