LoadBestModelCallback

class LoadBestModelCallback(folder, process_exclusion=False, monitor=None, larger_better=True, delete_after_train=True, **kwargs)

保存 monitor 值最佳的模型，并在训练结束的时候重新加载模型的 Callbcak。

默认会在加载之后删除权重文件。仅在训练正常结束的时候才能加载最好的模型。

参数:

• folder (str) – 保存的文件夹。

• process_exclusion (bool, default: False) – – 是否互斥地执行保存操作；在模型规模较大时该参数可以 节省一定的内存。

• monitor (Union[str, Callable, None], default: None) –

  监控的 metric 值。

  • 为 None 时，不设置监控值。

  • 为 str 时， CoLLiE 将尝试直接使用该名称从 evaluation 的结果中寻找，如果最终在 evaluation 结果中没有找到完全一致的名称，则将使用最长公共字符串算法 从 evaluation 结果中找到最匹配的那个作为 monitor。

  • 为 Callable 时， 则接受参数为 evaluation 的结果（字典类型），返回一个 float 值作 为 monitor 的结果，如果当前结果中没有相关的 monitor 值则返回 None。

• larger_better (bool, default: True) – 该 metric 值是否是越大越好；

• delete_after_train (bool, default: True) – 在训练结束后是否删掉模型；

• kwargs – 传给 Trainer.save_model() 和 Trainer.load_model() 的额外参数。