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

浙公网安备 33010602011771号