Skip to main content
 首页 » 编程设计

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

2024年11月01日32zhujiabin

我需要将使用 TensorFlow 1.12.0 Python 训练的模型转换为使用 TensorFlow.js 训练的模型。它兼容什么版本的 tf.js 和 tf.js 转换器?

请您参考如下方法:

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

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