(module, N, layer_share=False)
| 1545 | |
| 1546 | |
| 1547 | def _get_clones(module, N, layer_share=False): |
| 1548 | if layer_share: |
| 1549 | return nn.ModuleList([module for i in range(N)]) |
| 1550 | else: |
| 1551 | return nn.ModuleList([copy.deepcopy(module) for i in range(N)]) |
| 1552 | |
| 1553 | |
| 1554 | def build_transformer(args): |