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

Question

我正在训练 Mask R-CNN model。

我在 'head' 训练了 2 个时期后保存了权重，我想从第三个时期继续。但是 model.train() 函数没有 initial_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 个时期后，它的权重发生了变化。因此，当您再次调用 fit 时，模型将继续训练。你的进度不会丢失

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

How to start training from different epoch Mask R-CNN?

image-segmentation

deep-learning

keras

tensorflow