
tensorflow
Ooo。
让努力变成一种习惯
展开
-
Tensorflow实现网络模型剪枝--model_pruning模块
部分代码参考:https://blog.youkuaiyun.com/lai_cheng/article/details/90643100剪枝:剪枝就是利用某一个准则对某一组或某一个权值置0从而达到将网络神经元置0以达到稀疏化网络连接从而加快整个推理过程及缩小模型大小的迭代过程,这个准则有暴力穷尽组合排忧、使用对角 Hessian 逼近计算每个权值的重要性、基于一阶泰勒展开的模型代价函数来对权值排序、...原创 2020-03-31 17:38:23 · 3819 阅读 · 0 评论 -
tf-CNN(三)Tensorflow将训练模型保存到本地
训练过程在上一篇博客里https://blog.youkuaiyun.com/xue_csdn/article/details/105128094首先在代码的相同路径下新建一个文件夹save_model。模型会被保存进来。直接把网络结构和模型的训练和保存写在同一代码里。save_model.py"""###训练猫狗分类模型###保存生成的模型"""#import tensorfl...原创 2020-03-28 10:23:07 · 1667 阅读 · 0 评论 -
tf-CNN(二)Tensorflow制作自己的数据集并训练
参考:https://www.jianshu.com/p/15e3f74180fchttps://blog.youkuaiyun.com/wiinter_fdd/article/details/72835939对于数据量较小而言,可能一般选择直接将数据加载进内存,然后再分batch输入网络进行训练。但是,如果数据量较大,这样的方法就不适用了,因为太耗内存。在这儿我介绍一种比较通用,高效的读取方法,即使...原创 2020-03-26 21:06:54 · 4532 阅读 · 5 评论 -
tf-CNN(一)Tensorflow创建一个简单的CNN模型
参考:https://www.jianshu.com/p/e2f62043d02b利用tensorflow框架和Python语言编写一个简单的卷积神经网络结构CNN来识别手写数字(mnist数据集方便调用)网络一共包括4层,分别是卷积层conv1+池化pooling 卷积层conv2+池化pooling 全连接层fc1+dropout 全连接层fc1+softmax(预测)i...原创 2020-03-25 17:32:26 · 1824 阅读 · 0 评论 -
tensorflow1.x和tensorflow2.0两个版本的函数对比表
https://docs.google.com/spreadsheets/d/1FLFJLzg7WNP6JHODX5q8BDgptKafq_slHpnHVbJIteQ/edit#gid=0如果打不开网页,可以直接从我的网盘里下载:https://pan.baidu.com/s/1cWHtdQM1yoMOM4ZXt_vY9w,提取码:i953...原创 2019-11-05 20:36:51 · 17461 阅读 · 16 评论 -
Tensorflow-GPU安装所对应的cuda和cnDNN版本
参考tensorflow官网:https://www.tensorflow.org/install/source#common_installation_problemstf版本 Python 版本 编译器 编译工具 cuDNN CUDA tensorflow_gpu-1.13.1 2.7、3.3-3.6 GCC 4.8 Bazel 0.19...原创 2019-10-24 21:18:42 · 1200 阅读 · 0 评论