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