shipingge / mmer

Code for the paper Learning Robust Multi-Modal Representation for Multi-Label Emotion Recognition via Adversarial Masking and Perturbation

MMER

Code for the paper Learning Robust Multi-Modal Representation for Multi-Label Emotion Recognition via Adversarial Masking and Perturbation.

mmer's Issues

关于loss.backward(retain_graph=true)显示cuda memory error的问题

您好，我在运行代码到loss.backward(retain_graph=true)的时候，发现cuda memory error的问题，查找之后发现应该是计算图一直没有得到释放导致显存报错，就算将batch_size设置的很小也没有效果。在硬件设备有限的情况下，请问是否能将代码中的loss1.backward(retain_graph=true)和loss2.backward(retain_graph=true)改为total_loss = loss1 + loss2, total_loss.backward？这样虽然不会再报cuda memory error，但是我不知道这和您精巧设置的训练结构是否是等价的。期待您的回答！

关于NEMu数据集的使用

您好！很高兴接触到您的项目。想请问一下有没有NEMu数据集的文件可以提供下载？需要对数据集包进行什么样的处理？