RefineDet.PyTorch icon indicating copy to clipboard operation
RefineDet.PyTorch copied to clipboard

第一次forward的时候显存暴涨vgg

Open wuzuowuyou opened this issue 4 years ago • 0 comments

测试模式,当第一次运行到refinedet.py的forward的以下代码的时候:

for k in range(30):
            x = self.vgg[k](x)###############
            if 22 == k:
                s = self.conv4_3_L2Norm(x)
                sources.append(s)
            elif 29 == k:
                s = self.conv5_3_L2Norm(x)
                sources.append(s)

        # apply vgg up to fc7
        aaa = len(self.vgg)
        for k in range(30, len(self.vgg)):
            x = self.vgg[k](x)
        sources.append(x)

在 x = self.vggk这行,用vgg跑前向的时候, 当i=5的时候,执行x = self.vggk这句话,显存会快速达到4851M然后快速降到800多M 当i=7的时候,执行x = self.vggk这句话,显存会快速达到1609M然后快速降到800多M 。。。 看了下都是卷积层 但是为什么是第一次的时候,后面不会有问题了!!!!so,why第一次的时候会出现后面不会??

用的320大小的图片

wuzuowuyou avatar May 28 '20 03:05 wuzuowuyou