(self, config)
| 465 | |
| 466 | class BertEncoder(nn.Module): |
| 467 | def __init__(self, config): |
| 468 | super().__init__() |
| 469 | self.config = config |
| 470 | self.layer = nn.ModuleList([BertLayer(config,i) for i in range(config.num_hidden_layers)]) |
| 471 | self.gradient_checkpointing = False |
| 472 | |
| 473 | def forward( |
| 474 | self, |