哪个版本的TensorFlow.js与在TensorFlow 1.12.0(Python)中训练的模型兼容?

时间:2019-10-02 08:20:22

标签: python tensorflow tensorflow.js

我需要将在TensorFlow 1.12.0 Python中训练的模型转换为TensorFlow.js的模型。哪个版本的tf.js和tf.js转换器兼容?

1 个答案:

答案 0 :(得分:1)

您没有提到将模型保存在TensorFlow 1.12中的格式。我建议使用已保存的模型格式来保存您的模型。如果使用保存的模型,则可以使用最新版本的tf.js和tf.js转换器。 keras h5模型也是如此。

但是,如果将其保存为pb文件的形式,则必须使用0.15的tf.js版本和0.8.6的tf.js转换器

相关问题