(self, config)
| 553 | |
| 554 | class BertEncoder(nn.Module): |
| 555 | def __init__(self, config): |
| 556 | super().__init__() |
| 557 | self.config = config |
| 558 | self.layer = nn.ModuleList([BertLayer(config) for _ in range(config.num_hidden_layers)]) |
| 559 | self.gradient_checkpointing = False |
| 560 | |
| 561 | def forward( |
| 562 | self, |