yolov3-channel-and-layer-pruning icon indicating copy to clipboard operation
yolov3-channel-and-layer-pruning copied to clipboard

Published 20 hours ago • verified publisher icon tanluren

刚开始推理时，GPU内存占用持续增加

Open LiuRJun opened this issue 3 years ago • 0 comments

我的batch_size=4，GPU内存为6G. 在前向推理过程中，每次到了卷积层运行时，GPU内存就会增加，0.3G左右,调试发现，推理到第49层，GPU内存就爆了。 image 也就是在执行x=module(x)时就会增加GPU的内存占用，请问，有哪位大神能解释下原因

LiuRJun avatar Feb 18 '22 03:02 LiuRJun