def forward()

in models/regnet.py [0:0]


    def forward(self, x, *args, **kwargs):
        x = self.stem(x)
        x = self.trunk_output(x)
        if self.head is None:
            return x
        x = self.head(x)

        return x