2020
10-07
10-07
基于Keras 循环训练模型跑数据时内存泄漏的解决方式
在使用完模型之后，添加这两行代码即可清空之前model占用的内存：importtensorflowastffromkerasimportbackendasKK.clear_session()tf.reset_default_graph()补充知识：keras多个模型测试阶段速度越来越慢问题的解决方法问题描述在实际应用或比赛中，经常会用到交叉验证（10倍或5倍）来提高泛化能力，这样在预测时需要加载多个模型。常用的方法为mods=[]fromkeras.utils.generic_utilsimportCustomObjectScopew...
继续阅读 >