当Java和Python tensorflow版本都是1.2.0时,我们似乎可以使用SavedModelBundle(Java)和Saved Model API(Python)在Python tensorflow中保存经过训练的模型,并在Java tensorflow中加载模型(而不是使用Maven)。在
但是,当Python的版本低于1.0时,我无法找到一种在Java中正确加载模型的方法。在
我训练了一个模型,并将其保存为Python tensorflow(0.9.0)中的.pb、.sd和.txt文件,然后按照tensorflow网站中的example指令加载模型。但是,我得到了以下错误:Exception in thread "main" java.lang.IllegalStateException: Attempting
to use uninitialized value policy/mean_network/hidden_1/b
[[Node: _retval_policy/mean_network/hidden_1/b_0_0 =
_Retval[T=DT_FLOAT, index=0,
_device="/job:localhost/replica:0/task:0/cpu:0"]
(policy/mean_network/hidden_1/b)]]
at org.tensorflow.Session.run(Native Method)
at org.tensorflow.Session.access$100(Session.java:48)
at org.tensorflow.Session$Runner.runHelper(Session.java:285)
at org.tensorflow.Sessio