
tensorflow
imumu_xi
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
Alexnet网络模型在cifar-10数据集上的实现(基于tensorflow-gpu)
最近根据github和tensoflow源代码中的关于Alexnet的一些代码,完成了一个在cifar-10训练集上进行训练和测试的Alexnet模型(算是半抄半改吧,哈哈!!)。实现在测试集上的accuracy=74%左右。最近没空,我就简单写写。一,先总结我最近学习卷积神经网络的的大致过程1.先看周志华的《机器学习》这本教材,我是把第五章神经网络看完结束,剩下的以后有空再看。最好要把里面公...原创 2018-03-14 22:18:54 · 9104 阅读 · 21 评论 -
TensorFlow四种Cross Entropy算法实现和应用
直接上别人的链接:http://dataunion.org/26447.html转载 2018-03-17 10:36:48 · 313 阅读 · 0 评论 -
机器学习中代码出现tensorflow.python.framework.errors_impl.InternalError,from device: CUDA_ERROR_OUT_OF_MEMORY
此文为解决机器学习中使用tensorflow,在运行代码出现上述报错情况1.先运行nvidia-smi 检查GPU运行情况,若内存够用进入22.代码应作已下修改import tensorflow as tfimport osos.environ["CUDA_VISIBLE_DEVICES"] = '0' #use GPU with ID=0config = tf.ConfigPro...原创 2018-03-06 17:49:05 · 39366 阅读 · 7 评论 -
numpy 对矩阵,数组的操作
1.对矩阵每个元素求绝对值np.abs(W)2.对矩阵转置,假设我们矩阵A是四维的3*4*32*64,经过以下转置A=A.transpose(3,2,0,1)然后A就变成64*32*3*4的矩阵了3.矩阵求和,求平方temp=np.sum(A,(a,b...))将矩阵中每个元素变为其平方数 temp**2以下给个例子可以看出np.sum()的第二个参...原创 2018-03-24 15:11:06 · 4401 阅读 · 0 评论 -
一周bug(深度学习&&tensorflow)(一)
1在运行tensorflow时,发现GPU利用率很低,但是显存却占满了,而cpu却高达1600%的利用率,自己代码中也添加了使用GPU运行,但是并没有什么用经过检查发现: 我本来只是想在提取训练集与测试集图片时用CPU,在训练时用GPU,但是我在print语句后,所有代码都放在了with语句下,这样即使我在以后的训练代码部分加了GPU使用语句但是还是会默认使用cpu2在对损失加入正则项的时候出现t...原创 2018-03-30 20:30:36 · 545 阅读 · 0 评论 -
tensorflow对训练变量checkpoint的保存与读取
1.保存变量先创建(在tf.Session()之前)saversaver = tf.train.Saver(tf.global_variables(),max_to_keep=1) #max_to_keep这个保证只保存最后一次training的训练数据然后在训练的循环里面 checkpoint_path = os.path.join(Path, 'model.ckpt')saver.save...原创 2018-03-30 21:03:54 · 20818 阅读 · 0 评论 -
tensorflow指定GPU运行以及GPU资源分配的几种方式
1.在终端执行时设置使用哪些GPU(两种方式)(1) 如下(export 语句执行一次就行了,以后再运行代码不用执行)(2)如下2.代码中指定(两种方式)(1)import osos.environ["CUDA_VISIBLE_DEVICES"] = "1"(2)# Creates a graph.with tf.device('/gpu:1'): a = tf.constant([1.0...原创 2018-04-08 21:03:04 · 6850 阅读 · 0 评论 -
tensorflow 数据读取总结---(直接供给数据(feeding) 从文件中以管线形式读取数据 预加载数据)
主要参考tensorflow官方中文社区,链接:http://www.tensorfly.cn/tfdoc/how_tos/reading_data.html#AUTOGENERATED-file-formats,这个社区中中给的一些示例链接有的不对,可以到这个链接里面找:https://tensorflow.googlesource.com/tensorflow/+/master/tensorf...原创 2018-04-14 15:49:01 · 7545 阅读 · 2 评论 -
batch normalization 中的 moving_mean与moving_variance理解
batch normalization在训练部分代码时看到下面这一行update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS) with tf.control_dependencies(update_ops): train_op = optimizer.minimize(loss, global_step)**************...原创 2018-04-14 20:25:34 · 5617 阅读 · 1 评论