在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
modelargument to be a functional
Modelinstance, but got a subclass model instead.
有没有其他的方法来克隆一个tf.keras.Model
的子类的模型?
1条答案
按热度按时间aiazj4mn1#
这是不可能的,但在Keras存储库here上跟踪了该问题。