在每个时期后保存模型 - AllenNLP

Question

我是否可以在配置文件中设置一个参数（也许是为培训师设置的？），以便在每个时期或特定步数之后保存模型（存档）？ 我正在使用 seq2seq 数据加载器和“composed_seq2seq”作为我的模型。这是我的教练目前的样子：

"trainer": {
"num_epochs": 50,
"cuda_device": 0,
"optimizer": {
  "type": "adamw",
  "lr": 0.0005
}

Answer 1

您能否从每个 epoch/some 个步骤中详细解释一下您尝试对模型执行的操作？我认为它已经在每次获得新的最佳分数时对模型进行存档，所以我想知道你想要做什么而无法用它来完成。

编辑：

看起来AllenNLP已经在每个epoch保存了一个模型，但它默认最多只保留2个。我相信您可以通过在训练配置中添加 checkpointer 来更改它，例如：

  "trainer": {
    "checkpointer": {
      "num_serialized_models_to_keep": num_epochs
    },
  }