Getting an error Error(s) in loading state_dict for LCFS_ATEPC #314
-
|
I was running this code from pyabsa import AspectTermExtraction as ATEPC config.model = ATEPC.ATEPCModelList.LCFS_ATEPC I tried other models as welltrainer = ATEPC.ATEPCTrainer( I am getting the following error. RuntimeError Traceback (most recent call last) /opt/conda/lib/python3.7/site-packages/pyabsa/tasks/AspectTermExtraction/trainer/atepc_trainer.py in init(self, config, dataset, from_checkpoint, checkpoint_save_mode, auto_device, path_to_save, load_aug) /opt/conda/lib/python3.7/site-packages/pyabsa/framework/trainer_class/trainer_template.py in _run(self) /opt/conda/lib/python3.7/site-packages/pyabsa/tasks/AspectTermExtraction/instructor/atepc_instructor.py in run(self) /opt/conda/lib/python3.7/site-packages/pyabsa/framework/instructor_class/instructor_template.py in _train(self, criterion) /opt/conda/lib/python3.7/site-packages/pyabsa/framework/instructor_class/instructor_template.py in _resume_from_checkpoint(self) /opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py in load_state_dict(self, state_dict, strict) RuntimeError: Error(s) in loading state_dict for LCFS_ATEPC: |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 2 replies
-
|
remove the from_checkpoint arg in the trainer will fix this error |
Beta Was this translation helpful? Give feedback.
remove the from_checkpoint arg in the trainer will fix this error