save_checkpoint
接口功能
保存一个CheckPoint文件。
接口格式
save_checkpoint(checkpoint_name, model, optimizer, **kwargs)
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
|---|---|---|---|
checkpoint_name |
必选 |
保存路径。 |
有效的checkpoint文件路径。 |
model |
可选 |
要保存的模型参数checkpoint。与optimizer二选一填写。 |
- |
optimizer |
可选 |
分布式优化器对象。与model二选一填写。 |
- |
kwargs |
可选 |
若使用的链式优化器，需添加额外参数。 |
- |
使用样例
>>> from torch_mindio import CheckpointHelper >>> CheckpointHelper().save_checkpoint(checkpoint_name=checkpoint_name, optimizer=optimizer)
父主题： CheckpointHelper接口