def __init__(self, config, **kwargs):
    super().__init__(**kwargs)
    self.layer = [TFConvBertLayer(config, name=f'layer_._{i}') for i in range(config.num_hidden_layers)]
def __init__(self, config, **kwargs):
    super().__init__(**kwargs)
    self.layer = [TFConvBertLayer(config, name=f'layer_._{i}') for i in range(config.num_hidden_layers)]