- 博客(6)
- 收藏
- 关注
RSS订阅转载 记录：测试本机下使用 GPU 训练时不会导致内存溢出的最大参数数目
本机使用的GPU是GeForce 840M，2G显存，本机内存8G。试验时，使用vgg网络，调整vgg网络中的参数，使得使用对应的 batch_size时不会提示内存溢出。使用的是mnist数据集，图片尺寸是28×28。结果如下：# batch_size = 1# 参数总数：5946134# batch_size = 32# 参数总数：5...
2019-01-14 17:12:00
479
转载 记录：将图片数据生成 tfrecords 文件并在训练使用时读取
直接用别人的就行了：https://github.com/myCVs/GenTFRecords转载于:https://www.cnblogs.com/mbcbyq-2137/p/10260894.html
2019-01-12 20:16:00
216
转载 记录：EM 算法估计混合高斯模型参数
当概率模型依赖于无法观测的隐性变量时，使用普通的极大似然估计法无法估计出概率模型中参数。此时需要利用优化的极大似然估计：EM算法。在这里我只是想要使用这个EM算法估计混合高斯模型中的参数。由于直观原因，采用一维高斯分布。一维高斯分布的概率密度函数表示为：多个高斯分布叠加在一起形成混合高斯分布：其中：k表示一共有k个子分布，。为什么累加之和为1？因为哪怕是混...
2019-01-01 15:35:00
498
转载 记录：Ubuntu 18.04 安装 tensorflow-gpu 版本
狠下心来重新装了系统，探索一下 gpu 版本的安装。比较令人可喜的是，跟着前辈们的经验，还是让我给安装成功了。由于我是新装的系统，就像婴儿般纯净，所以进入系统的第一步就是安装 cuda，只要这个不出错，后面的几乎就不会出错了。首先，得先确定自己的显卡是否支持（这个一般支持，我这几年前的电脑的显卡都支持安装 cuda）。然后确定自己要安装的是什么版本的 TensorFlow，因为不同的...
2018-12-09 22:31:00
229
转载 记录：tf.saved_model 模块的简单使用（TensorFlow 模型存储与恢复）
虽然说 TensorFlow 2.0 即将问世，但是有一些模块的内容却是不大变化的。其中就有 tf.saved_model 模块，主要用于模型的存储和恢复。为了防止学习记录文件丢失或者蠢笨的脑子直接遗忘掉这部分内容，在此做点简单的记录，以便将来查阅。最近为了一个课程作业，不得已涉及到关于图像超分辨率恢复的内容，不得不准备随时存储训练的模型，只好再回过头来瞄一眼 TensorFlow ...
2018-11-30 16:15:00
1636
转载 记录：TensorFlow 中的 padding 方式
TensorFlow 中卷积操作和池化操作中都有一个参数 padding，其可选值有 ['VALID', 'SAME']。在 TensorFlow 文档中只是给出了输出张量的维度计算方式，但是并没有说明当 padding='SAME' 时，如何进行补零操作。其给出的输出张量的维度计算公式：VALID 方式：output_shape[i] = ceil((input...
2018-11-27 15:38:00
130
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人