python - 需要构建Keras子模型

Question

我正在创建一个继承自 keras 模型的 python 类。

class MyModel(tf.keras.models.Model):

    def __init__(self, size, input_shape):
        super(MyModel, self).__init__()
        self.layer = tf.keras.layers.Dense(size, input_shape=(input_shape,))

    def call(self, inputs):
        return self.layer(inputs)

model = MyModel(5, 30)
model.summary()

这给了我这个错误:

ValueError: This model has not yet been built. Build the model first by calling `build()` or calling `fit()` with some data, or specify an `input_shape` argument in the first layer(s) for automatic build.

如果我在创建模型后添加一行就修复了:

model = MyModel(5, 30)
model(tf.keras.layers.Input((30,)))
model.summary()

但这看起来并不是最好的方法。我该如何修复它？

Answer 1

您可以在构造函数中调用self.build()。

类似这样的事情:

class MyModel(tf.keras.models.Model):

    def __init__(self, size, input_shape):
        super(MyModel, self).__init__()
        self.layer = tf.keras.layers.Dense(size, input_shape=(input_shape,))
        self.build(input_shape)

    def call(self, inputs):
        return self.layer(inputs)

model = MyModel(2, (5, 30))
model.summary()