爆内存

[Mamaka2020]

您好，我的第二阶段训练老是爆内存，请问您是怎么多卡训练的？

[yuuy07]

他三张V100 32G的，我们得调低点参数才行

[YZ]

代码使用的是NVIDIA V100-32G GPU，其他型号的显卡出现显存不足的问题需要调整相关超参数才能成功运行哈。2卡V100-32G的samples_per_gpu=4，每张卡显存用了25个G多一点。您看看改一下相应的配置文件里的这个samples_per_gpu参数再运行看看显存情况。