the model.compile function is throwing an error TypeError: l2_normalize() got an unexpected keyword argument 'axis'. How to resolve this?