请问各位大佬是否遇到过下面这个报错:
File "/home/ai/vLLM_test/train_SFT.py", line 174, in
trainer.train()
File "/home/ai/anaconda3/envs/vllm_test/lib/python3.12/site-packages/transformers/trainer.py", line 2171, in train
return inner_training_loop(
^^^^^^^^^^^^^^^^^^^^
File "/home/ai/anaconda3/envs/vllm_test/lib/python3.12/site-packages/transformers/trainer.py", line 2531, in _inner_training_loop
tr_loss_step = self.training_step(model, inputs, num_items_in_batch)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ai/anaconda3/envs/vllm_test/lib/python3.12/site-packages/transformers/trainer.py", line 3667, in training_step
self.optimizer.train()
File "/home/ai/anaconda3/envs/vllm_test/lib/python3.12/site-packages/accelerate/optimizer.py", line 128, in train
return self.optimizer.train()
^^^^^^^^^^^^^^^^^^^^
'AdamW' object has no attribute 'train'
请问各位大佬是否遇到过下面这个报错:
File "/home/ai/vLLM_test/train_SFT.py", line 174, in
trainer.train()
File "/home/ai/anaconda3/envs/vllm_test/lib/python3.12/site-packages/transformers/trainer.py", line 2171, in train
return inner_training_loop(
^^^^^^^^^^^^^^^^^^^^
File "/home/ai/anaconda3/envs/vllm_test/lib/python3.12/site-packages/transformers/trainer.py", line 2531, in _inner_training_loop
tr_loss_step = self.training_step(model, inputs, num_items_in_batch)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ai/anaconda3/envs/vllm_test/lib/python3.12/site-packages/transformers/trainer.py", line 3667, in training_step
self.optimizer.train()
File "/home/ai/anaconda3/envs/vllm_test/lib/python3.12/site-packages/accelerate/optimizer.py", line 128, in train
return self.optimizer.train()
^^^^^^^^^^^^^^^^^^^^
'AdamW' object has no attribute 'train'