Yutao Yang
Results
1
comments of
Yutao Yang
llama_factory_vl/lib/python3.10/site-packages/trl/trainer/ppo_trainer.py中的第416行`dataset = self._remove_unused_columns(dataset)`会将labels标签舍弃，一个简单粗暴的方法是改成`dataset = dataset`