src/transformers/models/regnet/modeling_regnet.py/RegNetStage/forward
class RegNetStage:    def forward(self, hidden_state):
        hidden_state = self.layers(hidden_state)
        return hidden_state