- 博客(5)
- 收藏
- 关注
RSS订阅原创 tensorflow使用GPU训练时的显存占用问题
最近用GoogleNet （inception-v3）做fine-tune训练时，遇到了一个问题。当我选择单个GPU训练时候，服务器中所有的GPU显存都会被占满，如下图所示： 出现以上问题的原因是tensorflow在训练时默认占用所有GPU的显存。 可以通过以下方式解决该问题： 1、在构造tf.Session()时候通过传递tf.GPUOptions作为可选配置参数
2016-12-30 21:21:34
64857
12
原创 tensorflow模型相关
最近学习tensorflow，对其中的模型的导入导出一直比较困惑，因此花了些力气研究了一下，最后归纳整理到本篇博客。 变量保存和恢复 在tensorflow中，变量用来存储和更新参数。变量创建时可以赋予name和初始值，并且在执行模型的其他操作之前必须对变量进行初始化。比较简单的一个方法是添加一个对初始化所有变量的操作，在使用模型前先执行这个操作。比如： #添加变量初始化操作init
2016-12-01 21:16:27
1302
原创 神经网络的传播过程
转自 http://techeffigytutorials.blogspot.co.uk/2015/01/neural-network-illustrated-step-by-step.html 前向传播过程 I1,I2 = 输入 f() = 激活函数（Tanh(), Sigmoid()或者其他) W = 当前神经元的权重 Wb = 偏置权重 N = 当前神经元的
2016-10-14 11:43:12
696
2
原创 LeetCode刷题笔记
21. Merge Two Sorted Lists 该题可以看作是链表归并排序的其中一个步骤（合并两个已排序的链表），实质是对两个链表进行插入排序。比较两个链表当前节点的大小，中间节点指向较小节点，较小节点再指向下一个节点。可以用递归进行public class Solution { public ListNode mergeTwoLists(ListNode l1, ListNo
2016-07-15 21:22:36
377
原创 opencv3.1+vs2015+win764位配置
开发环境： 操作系统：Windows 7旗舰版 64位 IDE: Visual studio Community 2015 OpenCV： OpenCV 3.1 for windows（下载地址：http://opencv.org/） 下面开始进行配置 Step1 : 安装好VS 和 OpenCV Step2—opencv环境变
2016-03-22 16:02:13
2489
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人