qq_32172681的博客

1、保存模型model.save('**.h5') 将整个模型保存下来,以后直接载入模型与训练数据即可开始训练. 不用再定义网络和编译模型.(这种方法已经保存了模型的结构和权重,以及损失函数和优化器) 载入模型代码如下： from keras.models import load_model model.load_model('**.h5') 2、保存模型权重model.save...

1、“软batch”、梯度累计 我是用mask rcnn做分割，模型比较庞大，1080显卡最多也就能跑batch size=2，但又想起到batch size=64的效果，那可以怎么办呢？一种可以考虑的方案是，每次算batch size=2，然后把梯度缓存起来，32个batch后才更新参数。也就是说，每个小batch都算梯度，但每32个batch才更新一次参数。 我的需求是，SGD+Momen...