python - 如何使用中间输出保存/加载模型

Question

我在 Keras 中编写自动编码器:

inputs = Input((n_channels,))
l1 = Dense(40, activation="relu")(inputs)
l2 = Dense(19)(l1)
l3 = Dense(40, activation="relu")(l2)
training_layer = Dense(n_channels)(l3)
unify_layer = Model(inputs=inputs, outputs=l2)
training_layer = Model(inputs=inputs, outputs=training_layer)

我使用 training_layer 进行训练，使用 unify_layer 进行预测，所以当我在保存后继续学习时，我希望能够访问两个端点。

[根据 Marcin 的评论进行编辑]Model.save 只允许我保存一个模型。当我打电话时:

unify_layer.save("unify")
training_layer.save("training")

然后

unify_layer = load_model("unify")
training_layer = load_model("training")

两层不再链接，即当我训练 training_layer 时，unify_layer 未训练。

Answer 1

哦，我实际上可以使用save_weights 和load_weights 方法:

class Autoencoder():
    def __init__(self):
        inputs = Input((n_channels,))
        l1 = Dense(40, activation="relu")(inputs)
        l2 = Dense(19)(l1)
        l3 = Dense(40, activation="relu")(l2)
        training_layer = Dense(n_channels)(l3)
        self.unify_layer = Model(inputs=inputs, outputs=l2)
        self.training_layer = Model(inputs=inputs, outputs=training_layer)

    def save(self, filename):
        self.unify_layer.save_weights("unify_" + filename)
        self.training_layer.save_weights("training_" + filename)

    def load(self, filename):
        self.unify_layer.load_weights("unify_" + filename)
        self.training_layer.load_weights("training_" + filename)