kangmingyu

VGG16，显存溢出问题

6

config文件里面，我设置了参数batch_size = 32，normal_size = 224，然后使用VGG16之后，显存直接溢出了。训练集共3400多张图片，按说不至于吧。 我尝试了使用CPU + 内存运行，我显存8G，内存32G，想看下实际模型占用需要多少内存，结果一个batch之后，内存占用竟然达到31.1G。全部加载完图片大概是10G内存占用，也就是说中间21.1G是模型的。 我想确认下，请问up你有遇到过这个情况吗？或者有比较好的解释和解决方案吗？

I'm sorry to find the 'edm' package in the test_skccm.py of the directionary skccm . Can you reponse to me where is the edm package?