load_checkpoint

接口功能

从CheckPoint文件加载。

接口格式

load_checkpoint(checkpoint_name, optimizer, **kwargs)

接口参数

参数

是否必选

说明

取值要求

checkpoint_name

必选

加载路径。

有效的CheckPoint文件路径。

optimizer

可选

优化器实例。若为空,认为加载的是模型参数;若为有效优化器实例,认为加载的优化器参数。

-

kwargs

可选

若使用的链式优化器,需添加额外参数。

-

使用样例

>>> from torch_mindio import CheckpointHelper
>>> CheckpointHelper().load_checkpoint(filename, optimizer=optimizer)