python - 当两个模型同时对不同的数据进行训练时，如何将一个模型从一个模型传递到另一个模型？

Question

我有两个模型在两个不同的数据集上训练。但我想用不同的数据训练第二个模型，其初始点将是第一个模型的最后一个 dropout 层（附图显示了两个模型的架构）。我怎么能做到这一点？

模型架构

标签： pythonmachine-learningkeraslstmkeras-layer

Answer 1

不确定您正在使用什么框架，但我知道 Pytorch 可以做到这一点。不确定TensorFlow或其他。在训练您的第一个模型之后，您可以构建第二个模型的架构并从第一个模型中获取权重并逐层分配相应的层。如果您有一些代码示例，我可以进一步帮助您。假设您使用的是 Pytorch，您可以使用类似这里的东西。