检查点训练回调

class CheckpointTrainingCallback(schedule: str | CheckpointSchedule | type[CheckpointSchedule] | None = None, schedule_kwargs: Mapping[str, Any] | None = None, keeper: str | CheckpointKeeper | type[CheckpointKeeper] | None = None, keeper_kwargs: Mapping[str, Any] | None = None, root: Path | str | None = None, name_template: str = 'checkpoint_{epoch:07d}.pt')[来源]

基础类: TrainingCallback

在用户指定的周期保存检查点。

创建回调。

Parameters:

  • schedule (HintOrType[CheckpointSchedule]) – 选择检查点计划，参见 pykeen.checkpoints.scheduler_resolver

  • schedule_kwargs (OptionalKwargs) – 用于实例化检查点计划的关键字参数，如有必要， 参见 pykeen.checkpoints.scheduler_resolver

  • keeper (HintOrType[CheckpointKeeper]) – 选择检查点保留逻辑，参见 pykeen.checkpoints.keeper_resolver。 None 对应于保留所有创建的检查点。

  • keeper_kwargs (OptionalKwargs) – 基于关键字的参数，用于实例化保留策略（如果需要），参见 pykeen.checkpoints.keeper_resolver

  • root (pathlib.Path | str | None) – 检查点的根目录。默认为pykeen.constants.PYKEEN_CHECKPOINTS的一个新的子目录。

  • name_template (str) – 检查点文件的名称模板。可以包含一个格式键 {epoch}，该键将被实际的epoch替换。此回调不负责覆盖现有文件，即，如果您想保留多个检查点，请确保选择唯一的文件名。

方法总结

post_epoch(epoch, epoch_loss, **kwargs)

在epoch之后调用。

方法文档

post_epoch(epoch: int, epoch_loss: float, **kwargs: Any) → None[source]

在epoch之后调用。

Parameters:
Return type:

无