首页 > 解决方案 > 是否可以使用 Tensorflow 2.x 高级 API 在 Tensorflow 1.x 训练的模型上运行推理

问题描述

我已经将我的 Tensorflow 1.x 代码转换为 Tensorflow 2.x 并且我正在利用 TF 2.x 高级 API,但是我已经使用 TF 1.x 对我的模型进行了预训练,该模型由模型权重构成:

               checkpoint 
               snap-0.data-00000-of-00001 
               snap-0.meta 
               snap-0.index

我想用 TF 2.x 对模型进行推理,而不必创建会话和设置全局初始化程序。 我该怎么做呢? 我应该使用 TF 2.x API 重新训练模型并使用 TF 2.x 进行推理,还是可以使用 TF 2.x 高级 API 在 TF 1.x 训练的模型上运行推理

标签: pythontensorflowtraining-datainference

解决方案


推荐阅读