keras
zhe_csdn
不断犯错不断改错不断进步的过程
展开
-
tensorflow.python.framework.errors_impl.FailedPreconditionError: Attempting to use uninitialized val
tensorflow出现这样的错误是没有进行初始化。执行tensorflow的变量都必须进行先用init=tf.global_variables_initializer()sess.run(init)这样的代码进行初始化这种方法对我没有任何作用,我只要每调用一次API都会出现这样的错误,tensorflow1.15.2版本...原创 2020-04-28 14:14:35 · 1328 阅读 · 3 评论 -
使用keras框架训练神经网络时显卡内存使用很大很大的解决方案
修改源码:/usr/local/lib/python3.5/dist-packages/keras/backend/tensorflow_backend.pyelse:if _SESSION is None:if not os.environ.get(‘OMP_NUM_THREADS’):config = tf.ConfigProto(allow_soft_placement=True)...原创 2019-04-25 09:21:44 · 1921 阅读 · 0 评论 -
os.environ[]使用
os.environ[“CUDA_DEVICE_ORDER”] = “PCI_BUS_ID” os.environ[“CUDA_VISIBLE_DEVICES”] = “0”os.environ[“CUDA_DEVICE_ORDER”] = “PCI_BUS_ID” # 按照PCI_BUS_ID顺序从0开始排列GPU设备os.environ[“CUDA_VISIBLE_DEVICES”] = ...原创 2019-04-25 10:00:30 · 4180 阅读 · 0 评论 -
keras 指定显卡且限制现存用量(GPU/CPU使用)
keras在使用GPU的时候有个特点,就是默认全部占满显存。若单核GPU也无所谓,若是服务器GPU较多,性能较好,全部占满就太浪费了。于是乎有以下五种情况:1、指定GPU2、使用固定显存的GPU3、指定GPU + 固定显存4 GPU动态增长5 CPU充分占用一、 固定显存 GPU在使用keras时候会出现总是占满GPU显存的情况,可以通过重设backend的GPU占用情况来进行调...原创 2019-04-25 14:41:45 · 1617 阅读 · 0 评论 -
深度学习训练两种读取和训练方式差异
1、加载全部数据到显存中然后训练这种方式对于小数据量数据没问题,一次加注全部数据不会爆内存2、一个batch加载,加载一个完毕后训练再加载反反复复这种方式无论多少数据,只要设置batch合适都不会爆内存...原创 2019-04-25 15:04:32 · 808 阅读 · 2 评论 -
keras-yolov3 fine tune,OOM happens, add report_tensor_allocations
排除其他原因后,可能是这样的因为keras在训练时会一次占用完所有的内存,因此会修改源代码如下但是在微调训练时总会报OOM错误,因此需要将之前分配的显存空间注释系统中自带python环境,其位置在这里:源码地址:/usr/local/lib/python3.6/dist-packages/keras/backend/tensorflow_backend.pyanaconda 虚拟环境,其...原创 2019-05-06 11:44:52 · 728 阅读 · 0 评论