tensorflow
qq_32331347
这个作者很懒,什么都没留下…
展开
-
Tensorflow: 读取ckpt文件中的tensor
在使用pre-train model时候,我们需要restore variables from checkpoint files. 经常出现在checkpoint 中找不到”Tensor name not found”. 这时候需要查看一下ckpt中到底有哪些变量import osfrom tensorflow.python import pywrap_tensorflowc转载 2018-01-12 23:31:11 · 5857 阅读 · 1 评论 -
tensorflow选择性读取权重
在使用tensorflow加载网络权重的时候,直接使用tf.train.Saver().restore(sess, ‘ckpt’)的话是直接加载了全部权重,我们可能只需要加载网络的前几层权重,或者只要或者不要特定几层的权重,这时可以使用下面的方法:var = tf.global_variables()var_to_restore = [val for val in var if 'conv转载 2018-01-13 00:35:38 · 2090 阅读 · 0 评论