load_checkpoint
接口功能
加载CheckPoint文件。
接口格式
load_checkpoint(model, optimizer, opt_param_scheduler, load_arg='load', strict=True)
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
|---|---|---|---|
model |
必选 |
CheckPoint模型。 |
- |
optimizer |
必选 |
CheckPoint优化器。 |
- |
opt_param_scheduler |
必选 |
优化器参数调度器。 |
- |
load_arg |
可选 |
加载参数。 |
默认值："load"。 |
strict |
可选 |
是否严格加载所有模型。 |
默认值：True。 |
返回值
iteration
父主题： API接口参考