如何从不同时代的Mask R-CNN开始训练？

Question

我正在训练Mask R-CNN 模型.

我保存的权重后，训练了2个时代的 "头"，我想继续从三个时代。但是model.training()函数没有设置 初始epoch 参数作为序列模型的model.fit。

我有下面的代码，但如果我用加载的权重运行它，它就会从第一个纪元开始，我不希望这样。

EPOCHS = [1, 3, 5, 8]  

model.train(dataset_train, dataset_val,
                learning_rate = LEARNING_RATE,
                epochs = EPOCHS[1],
                layers = 'all',
                augmentation = augmentation)

如果有人能告诉我在我的情况下 initial_epoch 的替代品是什么，我会很感激。

Answer 1

在前2个epochs的拟合后，你的模型改变了它的权重。所以，当你调用 fit 再一次，模型将继续训练。你的进度不会丢失