search

zhangzx-uiuc / AMR-IE

The code repository for AMR guided joint information extraction model (NAACL-2021).
35 stars 8 forks source link

'Key error' on pretrained models when doing AMR parsing #9

Open ntu-zjy opened 1 year ago

ntu-zjy commented 1 year ago

I tried the code and pretrained model that you share, and met problems below. 2022-11-25 11:37:25 | INFO | fairseq.tasks.text_to_speech | Please install tensorboardX: pip install tensorboardX Traceback (most recent call last): File "testAMR.py", line 3, in <module> parser = AMRParser.from_checkpoint(in_checkpoint) File "/home/chenjingyuan/nlp/EE/TSAR-main/transition-amr-parser/transition_amr_parser/stack_transformer_amr_parser.py", line 242, in from_checkpoint model = load_models(args, task, use_cuda) File "/home/chenjingyuan/nlp/EE/TSAR-main/transition-amr-parser/transition_amr_parser/stack_transformer_amr_parser.py", line 32, in load_models task=task, File "/home/chenjingyuan/anaconda3/envs/transitionamr/lib/python3.7/site-packages/fairseq/checkpoint_utils.py", line 374, in load_model_ensemble state, File "/home/chenjingyuan/anaconda3/envs/transitionamr/lib/python3.7/site-packages/fairseq/checkpoint_utils.py", line 425, in load_model_ensemble_and_task state = load_checkpoint_to_cpu(filename, arg_overrides) File "/home/chenjingyuan/anaconda3/envs/transitionamr/lib/python3.7/site-packages/fairseq/checkpoint_utils.py", line 343, in load_checkpoint_to_cpu state = _upgrade_state_dict(state) File "/home/chenjingyuan/anaconda3/envs/transitionamr/lib/python3.7/site-packages/fairseq/checkpoint_utils.py", line 681, in _upgrade_state_dict state["cfg"] = convert_namespace_to_omegaconf(state["args"]) File "/home/chenjingyuan/anaconda3/envs/transitionamr/lib/python3.7/site-packages/fairseq/dataclass/utils.py", line 426, in convert_namespace_to_omegaconf _set_legacy_defaults(cfg.model, ARCH_MODEL_REGISTRY[args.arch]) KeyError: 'stack_transformer_6x6_nopos'

Jarvis658 commented 10 months ago

@whu-zjy I also had the same problem, did you solve it?

ntu-zjy commented 10 months ago

no, i do not know.

---Original--- From: @.> Date: Fri, Aug 25, 2023 18:34 PM To: @.>; Cc: "Jingyuan @.**@.>; Subject: Re: [zhangzx-uiuc/AMR-IE] 'Key error' on pretrained models when doingAMR parsing (Issue #9)

@whu-zjy I also had the same problem, did you solve it?

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>