Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Checkpoint not found for models #1

Closed
yuvalkry opened this issue May 24, 2024 · 1 comment
Closed

Checkpoint not found for models #1

yuvalkry opened this issue May 24, 2024 · 1 comment

Comments

@yuvalkry
Copy link

yuvalkry commented May 24, 2024

Hello,

Thank you for creating this example use of the layout parser!

Update: it is a problem in the layout parser:

         https://github.com/Layout-Parser/layout-parser/issues/168

How to load the models? there is a problem with caching, is it a bug in the model
or a problem with torch? how to resolve?

The file is actually there but with '?dl=1' addition

ls -lt ~/.torch/iopath_cache/s/57zjbwv6gh3srry/
total 836212
-rw-r--r-- 1 yuval yuval 856276388 May 24 08:40 'model_final.pth?dl=1'
-rw-r--r-- 1 yuval yuval 0 May 24 08:40 'model_final.pth?dl=1.lock'

Thanks,
       Yuval

config.yaml?dl=1: 8.19kB [00:01, 4.70kB/s]
model_final.pth?dl=1: 856MB [00:41, 20.8MB/s]
Traceback (most recent call last):
File "/mnt/c/Users/yuval/Documents/recommender/src/detectron Layout Parser.py", line 25, in
model1 = lp.Detectron2LayoutModel('lp://PubLayNet/mask_rcnn_X_101_32x8d_FPN_3x/config',
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/yuval/miniconda3/envs/recommender/lib/python3.11/site-packages/layoutparser/models/detectron2/layoutmodel.py", line 119, in init
self._create_model()
File "/home/yuval/miniconda3/envs/recommender/lib/python3.11/site-packages/layoutparser/models/detectron2/layoutmodel.py", line 122, in _create_model
self.model = detectron2.engine.DefaultPredictor(self.cfg)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/yuval/miniconda3/envs/recommender/lib/python3.11/site-packages/detectron2/engine/defaults.py", line 288, in init
checkpointer.load(cfg.MODEL.WEIGHTS)
File "/home/yuval/miniconda3/envs/recommender/lib/python3.11/site-packages/detectron2/checkpoint/detection_checkpoint.py", line 62, in load
ret = super().load(path, *args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/yuval/miniconda3/envs/recommender/lib/python3.11/site-packages/fvcore/common/checkpoint.py", line 153, in load
assert os.path.isfile(path), "Checkpoint {} not found!".format(path)
AssertionError: Checkpoint /home/yuval/.torch/iopath_cache/s/57zjbwv6gh3srry/model_final.pth not found!

@yuvalkry
Copy link
Author

Update: it is a problem in the layout parser:
Layout-Parser/layout-parser#168

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant