![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
keras
shuangyumelody
认真做好生活中的每一件小事
展开
-
Ubuntu-install-keras-cuda-tensorflow/theano
参考原网址:https://github.com/BIGBALLON/cifar-10-cnn/blob/master/doc/Ubuntu-install-cuda-tensorflow.md 特别声明:原网址上的大牛已经写的很详细了,我只是添加了自己遇到的问题的解决方案。STEP 1: Install Ubuntu 16.04到 Ubuntu官网 下载ISO镜像, 刻录进U盘进入Win10,转载 2017-10-27 21:56:55 · 591 阅读 · 0 评论 -
ResourceExhaustedError: OOM when allocating tensor with shape[10,256,400,528]
使用keras库进行模型训练时,出现以下错误:ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[10,256,400,528]解决方法如下:[10,256,400,528]的第一个参数表示batch_size的大小,第二个参数表示某层卷积核的个数,第三个参数表示图像的高,第四原创 2017-12-08 21:35:51 · 25334 阅读 · 17 评论 -
keras深度学习框架的训练保存及调用
#1.keras深度学习框架的训练保存过程from keras.models import model_from_jsonmodel_json = model.to_json()with open("model.json", "w") as json_file: json_file.write(model_json)# serialize weights to HDF5model.sa原创 2017-09-27 16:10:51 · 1182 阅读 · 0 评论 -
keras模型框架以图片的形式显示
原文网址:http://blog.csdn.net/u012313895/article/details/79075750首先安装keras2的三个可视化库sudo pip/pip3 install pydotsudo pip/pip3 install pydot_ngsudo apt-get install graphviz然后运行下面代码生成模型框架图fro转载 2018-01-19 21:04:29 · 1576 阅读 · 0 评论 -
keras实现多GPU或指定GPU的使用
1. keras新版本中加入多GPU并行使用的函数下面程序段即可实现一个或多个GPU加速: 注意:使用多GPU加速时,Keras版本必须是Keras2.0.9以上版本from keras.utils.training_utils import multi_gpu_model #导入keras多GPU函数import VGG19 #导入已经写好的函数模型,例如VGG19原创 2018-01-09 21:45:09 · 23422 阅读 · 5 评论