OOM

[Jesayy]

您好，请问您是在什么设备上进行训练的？我使用了3张A6000还是oom

[ZebangCheng]

我们使用的是A100（4卡*40G）。A6000一般是48G的显存，应该不会oom，您设置的batch size是多少？

[Jesayy]

batch size为1。请问是否可提供给详细的训练步骤和配置文件？

[ZebangCheng]

训练的话请参考minigpt4/datasets/datasets/first_face.py中的ECACEmotionDataset构造数据集。
训练的配置文件是train_configs/minigptv2_finetune_exp.yaml, 请注意关键词ECAC_emotion_caption, 你可以在项目中搜索ECAC_emotion_caption理清项目的执行逻辑。

如果你有其他问题，可以继续提问，我会一步一步回答。