def METHOD_NAME(self, x):
    for i in range(4):
        x = self.downsample_layers[i](x.contiguous())
        x = self.stages[i](x.contiguous())
    return x  # global average pooling, (N, C, H, W) -> (N, C)