Problems when running plot_parametric_plot.py

[Billy1900]

It seems like keras has changed the parameter of the function BatchNormalization. The error msg:

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.