小编Aja*_*tha的帖子

如果我需要使用与上面一行相同的代码，我通常会复制该行。我没有找到复制该行的快捷键。我看到一些帖子实现了重复行的快捷键，但它们相当复杂，并且每次都需要运行脚本。

我尝试使用以下方法保存和加载模型：所有键都已映射，但输出 #1 中没有预测

from detectron2.modeling import build_model
model = build_model(cfg)
torch.save(model.state_dict(), 'checkpoint.pth') 
model.load_state_dict(torch.load(checkpoint_path,map_location='cpu'))

我也尝试使用官方文档来做，但无法理解输入格式部分

from detectron2.checkpoint import DetectionCheckpointer
DetectionCheckpointer(model).load(file_path_or_url)  # load a file, usually from cfg.MODEL.WEIGHTS
checkpointer = DetectionCheckpointer(model, save_dir="output")
checkpointer.save("model_999")  # save to output/model_999.pth