MMCV中save_checkpoint

save_checkpoint是MMCV中用于保存模型参数和状态的函数。它将模型的权重、优化器状态和训练轮数等信息保存为一个pth文件。

该函数具有以下参数：

- filename：保存的文件名
- model：需要保存的模型
- optimizer：需要保存的优化器
- scheduler：需要保存的学习率调度器
- meta：元数据，包含训练轮数等信息
- create_symlink（可选）：是否创建一个符号链接指向最新的checkpoint文件

该函数的用法如下：

from mmcv.runner import save_checkpoint

save_checkpoint(
    filename,
    model,
    optimizer=None,
    scheduler=None,
    meta=None,
    create_symlink=True)

例如，以下代码演示了如何使用save_checkpoint来保存模型参数和状态：

from mmcv.runner import save_checkpoint

epoch = 10
model = build_model()
optimizer = build_optimizer(model)
scheduler = build_scheduler(optimizer)

checkpoint = {
    'model': model.state_dict(),
    'optimizer': optimizer.state_dict(),
    'scheduler': scheduler.state_dict(),
    'epoch': epoch,
}

save_checkpoint(checkpoint, 'checkpoint.pth')

运行上述代码会在当前目录下生成一个名为checkpoint.pth的文件，其中包含模型参数、优化器状态、学习率调度器状态和训练轮数等信息。