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!
