具有子类模型的Tensorflow克隆模型

2ul0zpep  于 2022-11-30  发布在  其他
关注(0)|答案(1)|浏览(164)

在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: Expectedmodelargument to be a functionalModelinstance, but got a subclass model instead.有没有其他的方法来克隆一个tf.keras.Model的子类的模型?

aiazj4mn

aiazj4mn1#

这是不可能的,但在Keras存储库here上跟踪了该问题。

相关问题