class detectron2.checkpoint.Checkpointer(model: torch.nn.modules.module.Module, save_dir: str = '', *, save_to_disk: bool = True, **checkpointables)
源代码:https://detectron2.readthedocs.io/_modules/fvcore/common/checkpoint.html#Checkpointer
基类:object
可以保存/加载模型以及其他可检查点对象的检查点。
__init__(model:torch.nn.modules.module.Module,save_dir:str ='',*,save_to_disk:bool = True,** checkpointables )
参数:
-
model(nn.Module):模型。
save_dir(str):保存和查找检查点的目录。
save_to_disk(bool):如果为True,则将检查点保存到磁盘,否则禁用此检查点的保存。
checkpointables(object):任何可检查点的对象,即具有state_dict()和load_state_dict()方法的对象。例如,它可以像 Checkpointer(model,"dir",optimizer =