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

Question

我正在训练 Mask R-CNN model .

我在“头部”训练 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 个 epoch 之后，改变了它的权重。所以，当您调用 fit模型将再次继续训练。你的进步不会丢失