
tensorflow
文章平均质量分 91
Mosay_dhu
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
读写tfreord文件——矩阵的存储和读取
先贴出常规的读写terecord文件的代码,按分类好的文件夹读取图片,并以图片所在的文件夹作为其对应的标签。import os import tensorflow as tf from PIL import Image #注意Image,后面会用到 IMAGE_SIZE = 224 #%% def _int64_feature(value): return tf.train...原创 2018-06-10 18:36:52 · 1364 阅读 · 3 评论 -
解决tf.train.MomentumOptimizer 发生NaN问题
之前一直用的的tensorflow的AdamOptimizer,由于想要复现下别人的论文,看到论文中的参数有momentum,因此需要用tf.train.MomentumOptimizer优化器,本来可以正常训练的代码再开始训练两步后就显示loss为NaN了。试了调整学习率和动量都没用。后来,到github上查了一下,发现有人提示修改如下.(参考链接https://github.com/tenso...原创 2018-10-11 14:49:18 · 5591 阅读 · 2 评论 -
Ubuntu18.04+RTX2080+cuda10+tensorflow
经历了两天的黑暗时光,终于把RTX2080成功装进了ubuntu中,mark一下。 从一开始装系统开始,我的主板型号是华硕ROG STRIX B360-I GAMING,算是很新的板子了,也导致了各种不兼容问题,比如之前装了ubuntu16.04,在ubuntu下无线网卡不能用,window下和ubuntu下声卡都不能用,关机关不掉。换了ubuntu18.04这些问题都解决了,建议最新的主板还是...原创 2018-10-18 21:36:13 · 36301 阅读 · 32 评论