Tensorflow 程序 内存泄漏+显存分配不足 数据预处理

    1. 目的                   在循环中,从硬盘中读取每个batch 的图片。     2. 错误步骤及原因                   图片截取从here。              3. 修改方法     方法一:可以将图片数据预处理部分不使用 tensor ,用的python 常见的numpy 等类型,这样就不会产生新的tensor 结点。     方法二:使
相关文章
相关标签/搜索