How to generate the capsule network codes using cubemx.ai

[Swu.4]

I create a capsule layer for my capsule network, as follows.

#A Capsule Implement with Pure Keras

class Capsule(Layer):

   def __init__...

   def build(....):

   def call(...):

...

model = Model(...)

model.compile(...)

model.fit(...)

model.save('***.h5')

But cubemx.ai reported errors while analyzing the model:  

INVALID MODEL: Couldn't load Keras model ***.h5,

[AI: network] Unknown layer: Capsule

Thanks a lot!