从CheckPoint文件加载。
load_checkpoint(checkpoint_name, optimizer, **kwargs)
参数 |
是否必选 |
说明 |
取值要求 |
---|---|---|---|
checkpoint_name |
必选 |
加载路径。 |
有效的CheckPoint文件路径。 |
optimizer |
可选 |
优化器实例。若为空,认为加载的是模型参数;若为有效优化器实例,认为加载的优化器参数。 |
- |
kwargs |
可选 |
若使用的链式优化器,需添加额外参数。 |
- |
>>> from torch_mindio import CheckpointHelper >>> CheckpointHelper().load_checkpoint(filename, optimizer=optimizer)