随笔分类 -  Tensorflow

摘要:这是tensorflow 一个经常性错误,错误的原因在于:显卡内存不够。 解决方法就是降低显卡的使用内存,途径有以下几种措施: 1 减少Batch 的大小 2 分析错误的位置,在哪一层出现显卡不够,比如在全连接层出现的,则降低全连接层的维度,把2048改成1042啥的 3 增加pool 层,降低整个 阅读全文
posted @ 2018-06-21 20:34 图米大侠 阅读(18993) 评论(1) 推荐(1)
摘要:1. tf.decode_raw(features['image_raw'],tf.uint8) 解码时,数据类型有没有错?tf.float32 和tf.uint8有没有弄混??? 2. tf.train.string_input_producer([data_file],num_epochs=1) 阅读全文
posted @ 2018-04-13 22:27 图米大侠 阅读(9195) 评论(1) 推荐(0)
摘要:tensorflow 当中的一个常用函数:Slice() def slice(input_, begin, size, name=None) 函数的功能是根据begin和size指定获取input的部分数据。 其中input是输入可以是列表,元祖等,begin 是起始地址,size是范围大小 如果是 阅读全文
posted @ 2018-03-06 16:42 图米大侠 阅读(426) 评论(0) 推荐(0)