(self, config)
| 393 | |
| 394 | class BertEncoder(nn.Module): |
| 395 | def __init__(self, config): |
| 396 | super().__init__() |
| 397 | self.config = config |
| 398 | self.layer = nn.ModuleList([BertLayer(config) for _ in range(config.num_hidden_layers)]) |
| 399 | |
| 400 | def forward( |
| 401 | self, |