- 博客(5)
- 收藏
- 关注
原创 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 2015OpenCV: OpenCV 3.1 for windows(下载地址:http://opencv.org/)下面开始进行配置Step1 :安装好VS 和 OpenCVStep2—opencv环境变
2016-03-22 16:02:13
2489
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人