sunwuhuang1-优快云博客

原创 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