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

Cannot load pretrained weight #11

Open
JasiRose opened this issue Apr 25, 2024 · 2 comments
Open

Cannot load pretrained weight #11

JasiRose opened this issue Apr 25, 2024 · 2 comments

Comments

@JasiRose
Copy link

Dear Mr. Zhang,
Firstly, I want to thank you for your efforts in developing and maintaining DIVS_Plus.

I'm writing to report an issue I encountered while trying to load the model weights from the repository. Specifically, when I tried to retrain Segmenter on the VSPW dataset, the weights for pretrained Segmenter Mask2Former(panoptic) ViT-L seem to be problematic. I've successfully loaded other weights provided in the repository, but when attempting to load this particular one, I receive the following error:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/root/anaconda3/envs/vspw/lib/python3.8/site-packages/torch/serialization.py", line 600, in load
    with _open_zipfile_reader(opened_file) as opened_zipfile:
  File "/root/anaconda3/envs/vspw/lib/python3.8/site-packages/torch/serialization.py", line 242, in __init__
    super(_open_zipfile_reader, self).__init__(torch._C.PyTorchFileReader(name_or_buffer))
RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

I've tried downloading the file multiple times from both baidupan and HuggingFace to rule out download corruption. Could you please check if the model weight file for Mask2Former(panoptic) ViT-L is uploaded correctly, or if there's a known issue with this file? Any guidance on how to resolve this issue would be greatly appreciated.

@zhang-tao-whu
Copy link
Owner

Sorry for the delayed response. It appears that the pre-trained weights are indeed corrupted. I'll need to retrain them, which may take some time.

@zhang-tao-whu
Copy link
Owner

zhang-tao-whu commented May 3, 2024

If you have some spare GPUs available, you can also try training it yourself. config

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

2 participants