Контрольная точка Detectron2 не найдена

Я получаю такую ​​ошибку с прошлой ночи. Обучил 5 моделей, проблем не было. Потом возникла такая проблема. Буду рад, если вы поможете.

AssertionError                            Traceback (most recent call last)
<ipython-input-9-08522bc16525> in <module>()
     34 os.makedirs(cfg.OUTPUT_DIR, exist_ok=True)
     35 trainer = CocoTrainer(cfg)
---> 36 trainer.resume_or_load(resume=False)
     37 trainer.train()

2 frames
/usr/local/lib/python3.6/dist-packages/fvcore/common/checkpoint.py in load(self, path, checkpointables)
    118         if not os.path.isfile(path):
    119             path = self.path_manager.get_local_path(path)
--> 120             assert os.path.isfile(path), "Checkpoint {} not found!".format(path)
    121 
    122         checkpoint = self._load_file(path)

AssertionError: Checkpoint https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x/139173657/model_final_68b088.pkl not found!

detectron
person Özgür Kan    schedule 16.12.2020    source источник

Ответы (2)


arrow_upward
2
arrow_downward

На данный момент я нашел ручное решение. Я скачал модель по ссылке и сохранил в своем профиле диска. Затем я скопировал модель из профиля диска в выходной файл и внес изменения в приведенный ниже код.

cfg.MODEL.WEIGHTS = /content/output/model_final_68b088.pkl

Может быть, есть лучшее решение, но я не знаю. Думаю, есть проблема с доступом к файлу модели по ссылке.

person Özgür Kan    schedule 16.12.2020

arrow_upward
1
arrow_downward

Просто смонтируйте файл рассола с диска Google в свое приложение.

cfg.MODEL.WEIGHTS = "/content/drive/MyDrive/file_model_weight/model_final_68b088.pkl"  # Let training initialize from model zoo

он не может получить доступ к файлу рассола модели из зоопарка модели, поэтому мы должны предоставить его локально или мы можем напрямую смонтировать его с диска Google.

person Prakhar Shukla    schedule 05.01.2021