我有两个模型，Tensorflow 2.0中的模型A和Pytorch 1.3中的模型B。模型A的输出就是B的输入。我希望端到端地训练这两个模型。有没有可能不把其中一个模型移植到另一个框架上呢？
bvjveswy1#
我相信这是不可能在Tensorflow和Pytorch中联合训练模型的，那两个框架使用非常不同的后端架构来计算损耗和做反向传播，所以它们在训练深度学习模型方面是不兼容的。更详细的问题应该是您在问题中使用的是哪个Tensorflow模型和哪个Pytorch。随着深度学习社区的发展，越来越多的基础深度学习算法都有不同版本的实现，并同时支持Pytorch和Tensorflow。很少会出现只能在Pytorch和Tensorflow中找到唯一实现的情况。只要尝试找到相应的实现并将它们结合在一起！
1条答案
按热度按时间bvjveswy1#
我相信这是不可能在Tensorflow和Pytorch中联合训练模型的，那两个框架使用非常不同的后端架构来计算损耗和做反向传播，所以它们在训练深度学习模型方面是不兼容的。
更详细的问题应该是您在问题中使用的是哪个Tensorflow模型和哪个Pytorch。随着深度学习社区的发展，越来越多的基础深度学习算法都有不同版本的实现，并同时支持Pytorch和Tensorflow。很少会出现只能在Pytorch和Tensorflow中找到唯一实现的情况。只要尝试找到相应的实现并将它们结合在一起！