keras
关注
分享
扫一扫
北方的杨先生
声明:博文的编写，主要参考网上资料，并结合个人见解，仅供学习、交流使用，如有侵权，请联系博主删除，原创文章转载请注明出处。博主微信：wx_big。
展开
-
Keras学习002：keras指定运行时显卡及限制GPU用量
keras在使用GPU的时候有个特点，就是默认全部占满显存。这样如果有多个模型都需要使用GPU跑的话，那么限制是很大的，而且对于GPU也是一种浪费。因此在使用keras时需要有意识的设置运行时使用那块显卡，需要使用多少容量。这方面的设置一般有三种情况：1.指定显卡 2.限制GPU用量 3.即指定显卡又限制GPU用量查看GPU使用情况语句（linux）1秒钟刷新一次watch ...原创 2018-11-14 10:15:11 · 1381 阅读 · 0 评论 -
Keras学习001：keras读取模型某一层的输出
keras提供很多训练好的模型vgg16,vgg19,InceptionV3等如何获取这些模型的某一层的输出 这些模型每个模型都有超过10个层，如果想检查中间的某层输出如何操作1. 加载模型以vgg16为例from keras.applications.vgg16 import VGG16model = VGG16(weights='imagenet', include_top=...原创 2018-11-14 10:15:41 · 10554 阅读 · 4 评论