keras
sirius_black_
这个作者很懒,什么都没留下…
展开
-
多GPU运行
转载自keras中文文档Docs » keras新手指南 » 常见问题与解答如何在多张GPU卡上使用Keras?我们建议有多张GPU卡可用时,使用TnesorFlow后端。有两种方法可以在多张GPU上运行一个模型:数据并行/设备并行大多数情况下,你需要的很可能是“数据并行”数据并行数据并行将目标模型在多个设备上各复制一份,并使用每个设备上的复制品处理整个数据集的不同部分数据。Keras在keras...转载 2018-01-17 12:47:04 · 1278 阅读 · 0 评论 -
版本 bug
1.代码 https://github.com/jacobgil/keras-dcgan/blob/master/dcgan.pydef generator_containing_discriminator(g, d): model = Sequential() model.add(g) d.trainable = False model.add(d) re...原创 2018-02-21 22:11:17 · 1287 阅读 · 0 评论 -
keras 自定义层
1. Layer重写的基本思路?Keras层的骨架。只需要实现三个方法即可:build(input_shape): 这是你定义权重的地方。这个方法必须设self.built = True,可以通过调用super([Layer], self).build()完成。build,这个函数用来确立这个层都有哪些参数,哪些参数是可训练的哪些参数是不可训练的。call(x): 这里是编写层的功能逻辑的地方。你...原创 2018-04-29 15:01:17 · 10965 阅读 · 1 评论 -
keras踩坑记——gpu和cpu结果不同
问题描述: 训练了大概6个epoch 模型在cpu上的loss开始出现nan,在gpu上表现正常经查找资料https://www.oschina.net/question/3683714_2266089 这位兄台和我遇到了同样的问题,然而他并没有解决https://www.zhihu.com/question/62441748这位兄台给出了一个解释,但是和我的模型问题应该不相同,因为我的RNN模型...原创 2018-05-08 11:20:42 · 11984 阅读 · 4 评论 -
keras中siamese的实现
近期由于项目需要学习了一下Siamese网络,并用keras简单实现,现总结两种写法如下:1.keras 序贯模型input_shape = (105, 105, 1)left_input = Input(input_shape)right_input = Input(input_shape)#build convnet to use in each siamese 'leg'c...原创 2018-07-17 16:43:52 · 7270 阅读 · 8 评论 -
keras 一些tips
1.显存占用问题由于tensorflow在训练时默认指定所有GPU的显存,使用tensorflow后端的keras亦如此注:虽然占用了所有GPU的显存,但实际使用只有指定的GPU。----------(占着不用) (1)禁用gpu import osos.environ["CUDA_VISIBLE_DEVICES"] = "-1" (2...原创 2018-07-17 20:02:46 · 507 阅读 · 0 评论 -
OOM in keras
问题描述:使用keras搭建siamese网络时,遇到错误如下: OOM when allocating tensor with shape[129024,4096] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [[Node: dense_1/kernel/Ass...原创 2018-07-14 12:47:57 · 8242 阅读 · 0 评论 -
keras中函数式模型示例
1、双输入单输出模型(如siamese)https://www.kaggle.com/kmader/image-similarity-with-siamese-networks2、单输入双输出https://baijiahao.baidu.com/s?id=1608764496267185255&wfr=spider&for=pc...原创 2018-08-17 10:20:35 · 1509 阅读 · 0 评论