提交异步任务,用于保存完本次迭代中的优化器参数和模型参数文件后更新跟踪器文件。
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 >>> )