跑示例代码出错

[xiaobo-lab]

以下是我的代码，安装的是pyabsa最新版本 from pyabsa import AspectTermExtraction as ATEPC from pyabsa import ModelSaveOption, DeviceTypeOption

config = ( ATEPC.ATEPCConfigManager.get_atepc_config_english() ) # this config contains 'pretrained_bert', it is based on pretrained models config.model = ATEPC.ATEPCModelList.FAST_LCF_ATEPC # improved version of LCF-ATEPC

dataset = ATEPC.ATEPCDatasetList.Restaurant14

config.batch_size = 16 config.patience = 2 config.log_step = -1 config.seed = [1] config.verbose = False # If verbose == True, PyABSA will output the model strcture and seversal processed data examples config.notice = ( "This is an training example for aspect term extraction" # for memos usage )

trainer = ATEPC.ATEPCTrainer( config=config, dataset=dataset, from_checkpoint="english", # if you want to resume training from our pretrained checkpoints, you can pass the checkpoint name here auto_device=DeviceTypeOption.AUTO, # use cuda if available checkpoint_save_mode=ModelSaveOption.SAVE_MODEL_STATE_DICT, # save state dict only instead of the whole model load_aug=False, # there are some augmentation dataset for integrated datasets, you use them by setting load_aug=True to improve performance )

在运行过程中发生以下错误，请问怎么有什么办法解决吗？ File "", line 1, in File "/usr/local/lib/python3.10/dist-packages/pyabsa/tasks/AspectTermExtraction/trainer/atepc_trainer.py", line 69, in init self._run() File "/usr/local/lib/python3.10/dist-packages/pyabsa/framework/trainer_class/trainer_template.py", line 241, in _run model_path.append(self.training_instructor(self.config).run()) File "/usr/local/lib/python3.10/dist-packages/pyabsa/tasks/AspectTermExtraction/instructor/atepc_instructor.py", line 794, in run return self._train(criterion=None) File "/usr/local/lib/python3.10/dist-packages/pyabsa/framework/instructor_class/instructor_template.py", line 357, in _train self._resume_from_checkpoint() File "/usr/local/lib/python3.10/dist-packages/pyabsa/framework/instructor_class/instructor_template.py", line 455, in _resume_from_checkpoint self.model.load_state_dict( File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 2041, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for FAST_LCF_ATEPC: Unexpected key(s) in state_dict: "bert4global.embeddings.position_ids".

[yangheng95]

请升级版本

[Koukotsukan]

去掉english的checkpoint