Importing model using Tf and Keras Error

[mendaxfz]

Importing the model using tensorflow and keras gives me the following error:

NameError Traceback (most recent call last)
in ()
----> 1 model = tf.keras.models.load_model('model.h5')

12 frames
/usr/local/lib/python3.6/dist-packages/tensorflow/python/keras/layers/core.py in (x)
680
681 def compute_output_shape(self, input_shape):
--> 682 input_shape = tensor_shape.as_shape(input_shape).as_list()
683 if not input_shape:
684 output_shape = tensor_shape.TensorShape([1])

NameError: name 'customPooling' is not defined

[ranasac19878]

I think you will need to save the customPooling layer as custom object while saving your model. See below link for help:
https://www.tensorflow.org/guide/keras/save_and_serialize