请问stage2 训练的数据集是？

[Limanman507]

请问stage2训练标签MERR_coarse_grained.txt对应的是哪个数据集？有没有stage 2训练的详细文档或者网页？

[ZebangCheng]

stage2训练标签MERR_coarse_grained.txt对应的是MER2023数据集。

stage 2训练的网页：
https://github.com/ZebangCheng/Emotion-LLaMA/blob/main/Instruction_Tuning.md

在之前的issues中也有一些关于微调的回答，可以参考。

[Limanman507]

感谢，目前能成功跑通stage1和2的训练，想请问是否在train_splits: ["train"]中加入类似["eval"]可以实现每训练一个epoch进行一次eval，然后保存checkpoint_best.pth，还是需要手动对每个epoch保存的checkpoint_xx.pth执行一次torchrun --nproc_per_node 1 eval_emotion.py --cfg-path eval_configs/eval_emotion.yaml --dataset feature_face_caption来手动比较哪个pth文件是best呢？

[ZebangCheng]

“是否在train_splits: ["train"]中加入类似["eval"]可以实现每训练一个epoch进行一次eval”是可行的。不过我们之前没有采取这样的方式，所以需要你们自己编写相关代码测试。

手动对每个epoch保存的checkpoint_xx.pth是我们使用的方案。