edgeyolo icon indicating copy to clipboard operation
edgeyolo copied to clipboard

Published 20 hours ago • verified publisher icon LSH9832

显存占用大

Open Chambers1994 opened this issue 2 years ago • 2 comments

训练时，大小接近的模型显存占用比yolox要大很多正常么？yolox可以bs=256，edgeyolo的bs=8

Chambers1994 avatar Feb 20 '23 08:02 Chambers1994

是的，本模型梯度信息更多，训练时显存占用较大。推理时不存在这种情况。

LSH9832 avatar Feb 20 '23 08:02 LSH9832

好的，谢谢

Chambers1994 avatar Feb 20 '23 08:02 Chambers1994