当两个模型同时训练不同的数据时，如何将一个模型的层传递给另一个模型？

Question

我有两个训练两个不同数据集的模型。但是我想用不同的数据训练第二个模型，其起始点将是第一个模型的最后一个辍学层（附图显示了两个模型的体系结构）。我将如何做到这一点？

model architecture

Answer 1

不确定您使用的框架是什么，但是我知道Pytorch可以实现。对TensorFlow或其他不确定。训练完第一个模型后，您可以构建第二个模型的体系结构，并从第一个模型中获取权重，并逐层分配各个层。如果您有一些代码示例，我可以为您提供进一步的帮助。假设您使用的是Pytorch，则可以使用here中的类似内容。