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

您好!对有线表格推理时加载模型失败 #152

Open
Selvaggiar opened this issue Jun 20, 2024 · 0 comments
Open

您好!对有线表格推理时加载模型失败 #152

Selvaggiar opened this issue Jun 20, 2024 · 0 comments

Comments

@Selvaggiar
Copy link

bash scripts/infer/demo_wired.sh 推理有线表格时,报了个关于加载模型权重的错误(但无线表格可以成功推理),请问该怎么解决呀?
Traceback (most recent call last):
File "demo.py", line 82, in
demo(opt)
File "demo.py", line 21, in demo
detector = Detector(opt)
File "/data/xyj/space/AdvancedLiterateMachinery/DocumentUnderstanding/LORE-TSR/src/lib/detectors/ctdet.py", line 32, in init
super(CtdetDetector, self).init(opt)
File "/data/xyj/space/AdvancedLiterateMachinery/DocumentUnderstanding/LORE-TSR/src/lib/detectors/base_detector.py", line 24, in init
self.model = create_model(opt.arch, opt.heads, opt.head_conv)
File "/data/xyj/space/AdvancedLiterateMachinery/DocumentUnderstanding/LORE-TSR/src/lib/models/model.py", line 31, in create_model
model = get_model(num_layers=num_layers, heads=heads, head_conv=head_conv)
File "/data/xyj/space/AdvancedLiterateMachinery/DocumentUnderstanding/LORE-TSR/src/lib/models/networks/pose_dla_dcn.py", line 493, in get_pose_net
head_conv=head_conv)
File "/data/xyj/space/AdvancedLiterateMachinery/DocumentUnderstanding/LORE-TSR/src/lib/models/networks/pose_dla_dcn.py", line 436, in init
self.base = globals()base_name
File "/data/xyj/space/AdvancedLiterateMachinery/DocumentUnderstanding/LORE-TSR/src/lib/models/networks/pose_dla_dcn.py", line 315, in dla34
model.load_pretrained_model(data='imagenet', name='dla34', hash='ba72cf86')
File "/data/xyj/space/AdvancedLiterateMachinery/DocumentUnderstanding/LORE-TSR/src/lib/models/networks/pose_dla_dcn.py", line 301, in load_pretrained_model
model_weights = model_zoo.load_url(model_url)
File "/data/xyj/anaconda3/envs/Lore/lib/python3.7/site-packages/torch/hub.py", line 595, in load_state_dict_from_url
return torch.load(cached_file, map_location=map_location)
File "/data/xyj/anaconda3/envs/Lore/lib/python3.7/site-packages/torch/serialization.py", line 713, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/data/xyj/anaconda3/envs/Lore/lib/python3.7/site-packages/torch/serialization.py", line 920, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '<'.

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