带子类模型的Tensorflow clone_model

Question

是否有办法在Tensorflow中克隆一个基于子类的模型?例如，如果我有以下模型:

class MySequentialModel(tf.keras.Model):
def __init__(self, name=None, **kwargs):
super().__init__(**kwargs)
self.dense_1 = FlexibleDense(out_features=3)
self.dense_2 = FlexibleDense(out_features=2)
def call(self, x):
x = self.dense_1(x)
return self.dense_2(x)

然后我训练保存，并加载模型，当我试图克隆它:

model = tf.keras.models.clone_model(original_model)

得到ValueError: Expected `model` argument to be a functional `Model` instance, but got a subclass model instead.是否有其他方法克隆一个模型作为tf.keras.Model的子类?

Answer 1

这是不可能的，但是这个问题在这里的Keras存储库中被跟踪。