tensorflow
MartianCoder
这个作者很懒,什么都没留下…
展开
-
Ubuntu16.04安装tensorflow
一:ubuntu16.04安装:1.镜像下载:链接:http://pan.baidu.com/s/1bpIQywJ 密码:87mv2.VMware Workstation Pro新建一个虚拟机,将下载好的镜像加载,创建一下。二:tensorflow安装参考:https://www.tensorflow.org/install/install_linux#prerequisite_pyt原创 2017-05-13 20:39:36 · 478 阅读 · 0 评论 -
Windows下安装Tensorflow
1.安装Pthon:https://www.python.org/downloads/release/python-352/注意修改路径直接安装在C:\Python35,方便使用在环境变量Path路径中增加:C:\Python35;C:\Python35\Scripts2安装Numpy:pip3 install Numpy3.安装Tensorflowpip3 install tensorf原创 2017-06-04 16:23:02 · 240 阅读 · 0 评论 -
TensorFlow模型文件保存和读取
一、模型文件的保存在训练一个TensorFlow模型之后,我们可以将训练好的模型保存成文件,这样可以方便下一次对新的数据进行预测的时候直接加载训练好的模型即可获得结果,下面通过TensorFlow提供的tf.train.Saver函数,将一个模型保存成文件,一般习惯性的将TensorFlow的模型文件命名为*.ckpt文件。[python] view plain copy<span styl...转载 2018-04-10 21:39:48 · 1749 阅读 · 0 评论 -
TensorFlow实现迁移学习程序调试问题汇总
1ValueError: The specified path: inception_v3.ckpt is a file. Please specify only the path prefix to the checkpoint files.问题描述:模型放在程序的同一文件路径中,但是就是找不到模型。出现问题的原因:tensorflow1.6.0版本中存在bug,参考资料https:...原创 2018-10-19 11:50:31 · 266 阅读 · 0 评论 -
调试AlexNet问题
1.from tensorflow.contrib.data import Dataset 加载不了解决:https://blog.csdn.net/kwame211/article/details/78579035在TensorFlow 1.3中,Dataset API是放在contrib包中的:tf.contrib.data.Dataset而在TensorFlow 1.4中,...原创 2018-10-19 11:51:50 · 460 阅读 · 0 评论 -
Ubuntu训练YOLO时出现:CUDA Error: out of memory
需要修改所使用的模型cfg文件中的subdivision的参数。由subdivisions=8改成subdivisions=64。subdivision:这个参数很有意思的,它会让你的每一个batch不是一下子都丢到网络里。而是分成subdivision对应数字的份数,一份一份的跑完后,在一起打包算作完成一次iteration。这样会降低对显存的占用情况。如果设置这个参数为1的话就是一...原创 2018-10-19 11:53:29 · 5569 阅读 · 2 评论