Tensorflow - 如何知道加载模型的配置,如重量形状,以便我可以提前创建该图形?

时间:2017-08-22 11:33:36

标签: tensorflow

我有不同的保存模型,其中一些使用不同的配置,即神经元的数量和优化器类型(Adam或RMS)。有没有办法在加载过程中了解这些参数,以便我可以更改图中的张量形状或优化器?感谢。

目前,我遇到了这个错误:

InvalidArgumentError (see above for traceback): Assign requires shapes of both 
tensors to match. lhs shape= [186,256] rhs shape= [186,32]

我知道我可以手动更改形状,但我想知道在代码中执行此操作的方法。此外,如果这是不可能的,有没有办法在将来保存模型,我可以提前知道他们的配置,以便我不会收到此错误?感谢。

0 个答案:

没有答案