1

我想制作一个 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 文件中？

4

0 回答 0