tensorflow
MarTin Guo
这个作者很懒,什么都没留下…
展开
-
GPU版TensorFlow怎么指定让CPU运行
由于某些原因GPU版的TensorFlow运行起来会出现一些问题,比如内存溢出等情况。此时我们可以用CPU和系统内存来运行我们的程序。 代码如下: import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 将以上三行代码放入程序首部即可。...原创 2018-07-25 10:05:41 · 9981 阅读 · 5 评论 -
TensorFlow只加载部分权重
var = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope="resnet_50/block1") saver = tf.train.Saver(var_list=var) saver.restore(sess, "resnet_50.ckpt") 以上代码段为resnet_50的网络只加载name_scope为(resnet_5...原创 2018-07-25 10:33:21 · 3598 阅读 · 0 评论 -
安装指定版本的tensorflow
例如安装1.4.0版本的tensorflow: pip install tensorflow-gpu==1.4.0原创 2018-08-24 21:09:19 · 2528 阅读 · 0 评论 -
numpy 如何读取.npy文件
weight = np.load("vgg16.npy", encoding="latin1").item()原创 2018-09-01 21:28:25 · 13755 阅读 · 9 评论 -
tensorflow怎么使用L-BFGS优化方法
L-BFGS优化方,可使用如下函数: self.Opt = contrib.opt.ScipyOptimizerInterface(self.loss, method="L-BFGS-B", options={'maxiter': 500, 'disp': 0}) self.Opt.minimize(self.sess, feed_dict={self.x: x, self.w: w}) ...原创 2018-09-10 09:38:09 · 6674 阅读 · 6 评论 -
tensorflow怎么查看.ckpt内保存的权重名和权重值
import os from tensorflow.python import pywrap_tensorflow checkpoint_path = os.path.join("C://Users//gmt//Desktop//rdb//", ".\\model.ckpt") # Read data from checkpoint file reader = pywrap_tensorflo...原创 2018-09-13 16:32:59 · 3682 阅读 · 1 评论 -
tf.control_dependencies的一点理解
先看一段代码:来源:https://blog.csdn.net/brooknew/article/details/80611165 import tensorflow as tf a = tf.Variable(2) selfAdd = tf.Variable(0) # selfAddition = tf.assign_add(selfAdd, 3) selfSub = tf.Variabl...原创 2018-10-31 20:26:56 · 750 阅读 · 0 评论