最近用tensorflow写了一个简单的神经网络,使用:
tf.save_model.simple_save
保存成SavedModel。然后想通过tensorflowjs转换成tfjs模型,命令如下:
tensorflowjs_converter --input_format=tf_saved_model
--output_format=tfjs_grapg_model
--signature_name=serving_default
--saved_model_tags=serve
./saved_model
./net_model
转换时出现错误:
tensorflow.python.framework.errors_impl.InvalidArgumentError: Failed to import metagraph, check error log for more info.
我查看了上面的error log ,发现错误在这:
2019-04-27 17:46:49.159440: E tensorflow/core/grappler/grappler_item_builder.cc:637] Init node W1/Assign doesn't exist in graph
其中的W1应该是我给hidden layer 1里面权重Weight命的名字。
找了很多地方都没有相关的错误,还望CSDN的各路大佬能帮帮萌新!谢谢!
我的运行环境是:Python3.6 tensorflow-gpu 1.13.1 tensorflowjs 1.0.1
再次感谢各位大佬的帮助!