Question

我使用keras库训练CNN模型，时间数为25.我可以在第一次使用10个时期运行模型，然后用这些代码行保存模型：

it "serializes the receipt" do
  expect(subject).to include('item-name' => 'ABC', 'item-price' => '100') # ... etc
end

然后我重新启动python并继续使用相同的数据集运行接下来的15个纪元，如下面的代码：

model.fit_generator(training_set,
                     steps_per_epoch = 100000,
                     epochs = 10,
                     validation_data = test_set,
                     validation_steps = 40000)
from keras.models import load_model
model.save('my_model.h5')

继续训练是否足够？或者我必须采取任何其他措施来继续这项工作。我非常感谢任何支持。

Answer 1

是的，没关系，{{1}}可以保存权重，模型架构和优化器状态，因此您可以毫无问题地恢复培训。

继续训练CNN与keras中保存的模型

1 个答案: