TypeError: ('Not JSON Serializable:'… when saving a model with custom layer ONLY in Google cloud platform

Question

I implemented a neural network consisting of a custom layer GlobalWeightedAveragePooling described below. The following error occurred when saving the model in Goog