TensorFlow
_rookie_coder
Talk is cheap.Show me the code!
展开
-
TensorFlow中的变量管理
本文参考《TensorFlow实战Google深度学习框架》一书,总结了一些在TensorFlow中与变量管理相关的一些API和使用技巧1.创建变量TensorFlow中可以通过tf.Variable和tf.get_variable两个函数来创建变量,两者基本功能相同,但是用法存在差别。#下面两个定义是等价的,只不过变量的名称不同v1 = tf.Variable(tf.constant(1.0,原创 2018-01-18 19:05:46 · 349 阅读 · 0 评论 -
TensorFlow中的模型持久化
本文参考《TensorFlow实战Google深度学习框架》一书,总结了一些在TensorFlow在保存训练好的模型过程中使用到的一些APITF提供了tf.train.Saver类来保存和还原一个神经网络模型1.模型保存模型保存的代码如下所示:先声明一个tf.train.Saver对象saver,然后使用saver.save进行保存,该函数的第二个参数是保存的路径。注意保存的文件名后缀为.ckpt原创 2018-01-18 22:39:12 · 474 阅读 · 0 评论 -
TensorFlow学习3:tf.gradients和tf.stop_gradient()
本文转载自http://blog.csdn.net/u012436149/article/details/53905797gradienttensorflow中有一个计算梯度的函数tf.gradients(ys, xs),要注意的是,xs中的x必须要与ys相关,不相关的话,会报错。 代码中定义了两个变量w1, w2, 但res只与w1相关#wrongimport tensorflow as tf...转载 2018-03-09 22:03:22 · 837 阅读 · 0 评论 -
TensorFlow笔记4:显卡使用
tensorflowtensorflow可以指定训练使用的显卡。如果一台电脑具有多个NVIDIA的GPUs,用户想要指定需要使用的GPU,那么在python中可以写如下语句。这个语句设置了当前程序可见的显卡。import osos.environ["CUDA_VISIBLE_DEVICES"]="0"然后使用tf.device来指定训练的gpu ID。但是如果只指定ID,而没有屏蔽掉...原创 2018-06-07 21:25:01 · 2750 阅读 · 0 评论