tf模型无法加载到tfjs

时间:2019-11-18 02:08:34

标签: tensorflow.js tensorflowjs-converter

我已经创建了一个模型,并且效果很好。后来我保存了以下命令。

export_path = os.getcwd()+"\\saved_models\\{}".format(int(t))
model.save(export_path, save_format='tf')

并使用以下命令将其转换为tfjs

tensorflowjs_converter --input_format=tf_saved_model ./saved_models/1574041992 ./saved_models/tfjs_model/

tfjs_model已创建,但是无法将其加载到tfjs中。我下载了一个样本模型,它运行良好。 当我比较两个model.json;

工作版本以以下内容开头:

{"format": "graph-model"

非工作版本以以下内容开头:

{"modelTopology": {"training_config": ...............

为什么我的模特与众不同?我是用错误的格式创建的吗?如果是,正确的格式是什么?

非常感谢您的回答。

TensorFlow.js版本

2.0.0

浏览器版本

版本78.0.3904.97(正式版本)(64位)

Python: 3.8

1 个答案:

答案 0 :(得分:0)

请仔细检查您的tensorflowjs pip版本,最新版本为1.3.2 您可以使用以下命令来更新点子:

pip install tensorflowjs --upgrade