`from keras.layers.normalization import BatchNormalization ` could be changed to `from tensorflow.keras.layers import BatchNormalization`