Keras version 2.4 changes

[hummingbird1989]

Thanks for the great work! I'm trying to get this to work with Keras 2.4 and encountering a few issues. The current one is as follows:

File "C:\Users\__\Anaconda3\envs\ocr\lib\site-packages\tensorflow\python\framework\tensor_util.py", line 463, in make_tensor_proto if shape is not None and np.prod(shape, dtype=np.int64) == 0: File "<__array_function__ internals>", line 6, in prod File "C:\Users\__\Anaconda3\envs\ocr\lib\site-packages\numpy\core\fromnumeric.py", line 3000, in prod keepdims=keepdims, initial=initial, where=where) File "C:\Users\__\Anaconda3\envs\ocr\lib\site-packages\numpy\core\fromnumeric.py", line 87, in _wrapreduction return ufunc.reduce(obj, axis, dtype, out, **passkwargs) TypeError: int() argument must be a string, a bytes-like object or a number, not 'NoneType'

This is happening due to a change in add_weight in FixedBatchNormalization.py where instead of passing shape, I pass (shape=shape)

[araj6947]

Could you please say about the inference time you encountered?