search

LIN-SHANG / InstructERC

The offical realization of InstructERC
121 stars 7 forks source link

不能同时设定emotion_prediction='True' 和speaker_task='True' #7

Open qftie opened 9 months ago

qftie commented 9 months ago

image 您好，从代码看好像不能同时设定emotion_prediction='True' 和speaker_task='True'，实际使用时也会报错找不到data目录。

graciechen commented 9 months ago

image 您好，从代码看好像不能同时设定emotion_prediction='True' 和speaker_task='True'，实际使用时也会报错找不到data目录。

看论文中“In the first stage of joint training, we use the speaker identification task for LLM In the second stage, we fine-tune LLM using both the ERC main task and the emotion influence prediction task to improve”所以第一阶段是speaker_task='True' ，第二阶段是emotion_prediction='True'和ERC main task联合，所以是不需要同时设定emotion_prediction='True' 和speaker_task='True'的

qftie commented 9 months ago

image 您好，从代码看好像不能同时设定emotion_prediction='True' 和speaker_task='True'，实际使用时也会报错找不到data目录。

看论文中“In the first stage of joint training, we use the speaker identification task for LLM看论文中“在联合训练的第一阶段，我们使用LLM的说话人识别任务 In the second stage, we fine-tune LLM using both the ERC main task and the emotion influence prediction task to improve”所以第一阶段是speaker_task='True' ，第二阶段是emotion_prediction='True'和ERC main task联合，所以是不需要同时设定emotion_prediction='True' 和speaker_task='True'的

您好，请问下如何设置参数实现以下两种方式训练呢？

  1. 不做speaker任务，直接emotion_prediction='True'和ERC main task联合
  2. 先做speaker任务，再emotion_prediction='True'和ERC main task联合训练