保存然后再利用CNN模型 - 保存初始化



我希望通过使用具有相同的超级参数的同一CNN重复使用相同的CNN来重复一系列图像分类实验,尤其是初始化。因此,如果我在实例化和训练之前保存模型,也可以保存初始化,以便稍后重新加载它,并使用不同的数据集和标签进行训练,它是否可以使用相同的超参数启动此新模型初始化是我使用第一个数据集/分类标签训练的第一个模型?我目前正在使用Fastai,当然是库的API库/集合,但我认为每个人都会有一个更一般性的解释,该解释涵盖了所有CNN使用任何库。

我希望答案说:"在工作流程创建CNN之后,该模型是初始化的,如果此时保存,则可以稍后重新加载并在下一个模型中使用相同的超参数和初始化。"

您可以在创建后立即保存学习者。

示例:

learn = cnn_learner(data,models.resnet34,metrics=error_rate)
learn.save('init')

稍后:

learn.load('init)

最新更新