yolov8改了一点为啥24G显存暴了

Question

有人知道yolov8n用4090训练可以一个小时训练完，但改进c2f后参数量才多了50万，训练时间直接干到1天，主要是24G显存暴了，batch只能设成4才行，而且我换了L20 48G显存还是照样暴显存，有人知道为啥吗

Answer 1

优化模型结构或减小batch size。

以上回复有部分引自免费微信小程序：皆我百晓生。 如问题尚未解决，还请稍待，如有疑问请回复进一步沟通。