FlagEmbedding icon indicating copy to clipboard operation
FlagEmbedding copied to clipboard

微调bge的时候报错,求助~

Open passionate11 opened this issue 9 months ago • 1 comments

报错如下 Traceback (most recent call last): File "/opt/tiger/ecom_Rag/FlagEmbedding/baai_general_embedding/finetune/run.py", line 222, in main() File "/opt/tiger/ecom_Rag/FlagEmbedding/baai_general_embedding/finetune/run.py", line 211, in main trainer.train() File "/usr/local/lib/python3.9/dist-packages/transformers/trainer.py", line 1885, in train return inner_training_loop( File "/usr/local/lib/python3.9/dist-packages/transformers/trainer.py", line 2147, in _inner_training_loop self.control = self.callback_handler.on_train_begin(args, self.state, self.control) File "/usr/local/lib/python3.9/dist-packages/transformers/trainer_callback.py", line 454, in on_train_begin return self.call_event("on_train_begin", args, state, control) File "/usr/local/lib/python3.9/dist-packages/transformers/trainer_callback.py", line 498, in call_event result = getattr(callback, event)( File "/usr/local/lib/python3.9/dist-packages/transformers/integrations/integration_utils.py", line 842, in on_train_begin self.setup(args, state, model, **kwargs) File "/usr/local/lib/python3.9/dist-packages/transformers/integrations/integration_utils.py", line 815, in setup model.save_pretrained(temp_dir) File "/usr/local/lib/python3.9/dist-packages/torch/nn/modules/module.py", line 1695, in getattr raise AttributeError(f"'{type(self).name}' object has no attribute '{name}'") AttributeError: 'BiEncoderModel' object has no attribute 'save_pretrained' 请问这是什么问题

passionate11 avatar May 19 '24 04:05 passionate11