在 Tensorflow 和 Pytorch 中联合训练模型

Question

我有两个模型，Tensorflow 2.0 中的模型 A 和 Pytorch 1.3 中的模型 B。模型 A 的输出是 B 的输入。我想端到端地训练这两个模型。

是否可以不将其中一个模型移植到另一个框架？

Answer 1

我认为在 Tensorflow 和 Pytorch 中联合训练模型是不可能的。这两个框架使用非常不同的后端架构来计算损失并进行反向传播，因此它们在训练深度学习模型时彼此不兼容。

更详细的问题应该是您在问题中使用哪种 Tensorflow 模型和哪种 Pytorch。随着深度学习社区的发展，越来越多的基础深度学习算法有各种版本的实现，并且同时支持Pytorch和Tensorflow。很少会出现只能在 Pytorch 和 Tensorflow 中找到独特实现的情况。只要尝试找到相应的实现并将它们连接在一起即可！