async_write_tracker_file

接口功能

提交异步任务,用于保存完本次迭代中的优化器参数和模型参数文件后更新跟踪器文件。

接口格式

async_write_tracker_file(iteration, iteration_dir, total_file_count, tracker_filename, callback)

接口参数

参数

是否必选

说明

取值要求

iteration

必选

当前处于第几次迭代。

有效的迭代次数。

iteration_dir

必选

此迭代的完整目录路径。

有效的路径。

total_file_count

必选

预计为此迭代保存的检查点文件的总数。

预估的文件数应与实际文件数相等。

tracker_filename

必选

tracker file的完整路径。

有效的跟踪文件路径。

callback

可选

异步任务完成时调用的回调函数。原型为:callback(iteration: int, result: int),result = 0表示成功,否则表示失败。

有效的可调用函数。

使用样例

>>> from torch_mindio import CheckpointHelper
>>> CheckpointHelper().async_write_tracker_file(
>>>     iteration=iteration,
>>>     iteration_dir=checkpoint_path,
>>>     total_file_count=file_count,
>>>     tracker_filename=tracker_file_path
>>> )