如何读取张量流中的所有参数?

时间:2017-03-19 06:57:39

标签: python tensorflow checkpoint

我想创建一个包含每个图层的权重和偏差的txt文件。 所以我试图从检查点文件中提取这些参数,如下所示

....
log = open("/home/extract_weights/weight.txt","w")
print(tf.contrib.framework.load_variable('/home/extract_weights/weight',"Variable_4/Adam"),file=log)
log.close()

之后,当我打开weight.txt文件时,它不显示'Variable_4 / Adam'的所有参数。因为'Variable_4 / Adam'有如此多的参数,它会将一些参数显示为'...,',如下所示。

[[[ -3.28861130e-03  -4.36393253e-04   2.71450588e-03 ...,
     -1.02883030e-03   7.19045638e-04   2.01382302e-03]
   [ -2.78162770e-04  -9.16915829e-04   6.41513802e-03 ...,
     -1.56752067e-05   1.02308008e-03   4.38673422e-03]
   [ -2.39563966e-03  -1.86417182e-03   6.60661049e-03 ...,
     -1.80270139e-03  -1.77437195e-03   6.22929214e-03]
   ...,
   [ -5.03436942e-03  -7.72555824e-04   4.88848845e-03 ...,
     -1.42029661e-03  -2.02224823e-03   6.00241497e-03]
   [ -3.24321515e-03   7.12389883e-05   3.92327132e-03 ...,
     -5.55949169e-04   6.48389920e-04   4.28168190e-04]
   [ -2.34445534e-03  -1.03506667e-03   5.81974722e-03 ...,
     -1.64756668e-03  -8.65898794e-04   6.68829679e-03]]

  [[ -5.87272050e-04  -6.52794901e-04   5.46910102e-03 ...,
     -2.80088483e-04  -1.18806446e-03   2.98094237e-03]
   [  1.95095991e-03  -1.39068766e-03   7.42691476e-03 ...,
    -5.30915044e-04  -6.42847299e-05   1.92967139e-03]
   [ -6.43472234e-03  -3.83679318e-04   3.45606497e-03 ...,
    -5.63960406e-04  -1.46852643e-03   5.95639553e-03]
   ...,

如何在txt文件中保存每个图层的所有参数?

0 个答案:

没有答案