AssertionError: Checkpoint /home/yourstorePath/model_final_5bd44e.pkl not found!

明泽.

于 2021-04-13 17:32:54 发布

阅读量2.4k

点赞数

文章标签： python

本文链接：https://blog.youkuaiyun.com/qq_36955294/article/details/115673898

版权

Detectron2 模型训练权重文件 checkpoint resume_or_load

关键词由优快云通过智能技术生成

本文档解析了在使用Detectron2进行模型训练时遇到的'Checkpoint not found!'错误，重点在于如何通过调整cfg.MODEL.WEIGHTS路径来解决找不到预训练模型的问题，适合初学者参考。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Traceback (most recent call last):
File "train.py", line 304, in <module>
    launch(
File "/10t/lvying/miniconda3/envs/detectron2/lib/python3.8/site-packages/detectron2/engine/launch.py", line 62, in launch
    main_func(*args)
File "train.py", line 293, in main
    trainer.resume_or_load(resume=args.resume)
File "/10t/lvying/miniconda3/envs/detectron2/lib/python3.8/site-packages/detectron2/engine/defaults.py", line 353, in resume_or_load
    checkpoint = self.checkpointer.resume_or_load(self.cfg.MODEL.WEIGHTS, resume=resume)
File "/10t/lvying/miniconda3/envs/detectron2/lib/python3.8/site-packages/fvcore/common/checkpoint.py", line 215, in resume_or_load
    return self.load(path, checkpointables=[])
File "/10t/lvying/miniconda3/envs/detectron2/lib/python3.8/site-packages/fvcore/common/checkpoint.py", line 141, in load
    assert os.path.isfile(path), "Checkpoint {} not found!".format(path)
AssertionError: Checkpoint /home/yourstorePath/model_final_5bd44e.pkl not found!

解决办法：cfg.MODEL.WEIGHTS