FoundationVision / GLEE

[CVPR2024 Highlight]GLEE: General Object Foundation Model for Images and Videos at Scale
https://glee-vision.github.io/
MIT License
1.02k stars 82 forks source link

RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory #32

Open stihuangyuan opened 3 months ago

stihuangyuan commented 3 months ago

GLEEmodel_swin load error, how to solve this?

Traceback (most recent call last): File "test_demo.py", line 102, in GLEEmodel_swin = GLEE_Model(cfg_swin, None, device, None, True).to(device) File "/data/huangyuan/GLEE_ws/GLEE/projects/GLEE/glee/models/glee_model.py", line 59, in init self.backbone = build_backbone(cfg) File "/data/huangyuan/GLEE_ws/GLEE/detectron2/modeling/backbone/build.py", line 31, in build_backbone backbone = BACKBONE_REGISTRY.get(backbone_name)(cfg, input_shape) File "/data/huangyuan/GLEE_ws/GLEE/projects/GLEE/glee/backbone/swin.py", line 738, in init self.init_weights(pretrained_weight) File "/data/huangyuan/GLEE_ws/GLEE/projects/GLEE/glee/backbone/swin.py", line 652, in init_weights checkpoint = torch.load(pretrained, map_location='cpu') File "/usr/local/lib/python3.8/dist-packages/torch/serialization.py", line 1004, in load with _open_zipfile_reader(opened_file) as opened_zipfile: File "/usr/local/lib/python3.8/dist-packages/torch/serialization.py", line 456, in init super().init(torch._C.PyTorchFileReader(name_or_buffer)) RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory