"显存"搜索结果 1 条

深度学习训练模型时,GPU显存不够怎么办?

近期发布的MMEngine v0.8.3版本中,包含了一项新功能,只需要在运行时加入一段命令行参数,MMDetection等下游代码即可 无损节省20%显存。用户只需要在训练脚本中加入--cfg-options efficient_conv_bn_eval="[backbone]"启动参数,即可启用该功能。下面是该功能在detection、classification任务上的加速效果实测,该功能在表格中称为“Tune Mode”。 这项功能的原理是什么?对什么样的模型有效果?如何使用这项功能?本文将从以上…