Keras
文章平均质量分 77
DexterLeiX
这个作者很懒,什么都没留下…
展开
-
[Keras] 使用多 gpu 并行训练并使用 ModelCheckpoint() 可能遇到的问题
问题描述在使用 callbacks.ModelCheckpoint() 并进行多 gpu 并行计算时,callbacks 函数会报错:TypeError: can't pickle ...(different text at different situation) objects这个错误形式其实跟使用多 gpu 训练时保存模型不当造成的错误比较相似: To save th...原创 2018-05-18 17:43:45 · 7250 阅读 · 0 评论 -
[Keras] SGD 随机梯度下降优化器参数设置
随机梯度下降Keras 中包含了各式优化器供我们使用,但通常我会倾向于使用 SGD 验证模型能否快速收敛,然后调整不同的学习速率看看模型最后的性能,然后再尝试使用其他优化器。Keras 中文文档中对 SGD 的描述如下: keras.optimizers.SGD(lr=0.01, momentum=0.0, decay=0.0, nesterov=False) 随机梯度下降法,支...原创 2018-05-15 10:56:22 · 44989 阅读 · 5 评论